Window10上Tensorflow的安装(CPU和GPU版本)


Posted in Python onDecember 15, 2020

之前摸索tensorflow的时候安装踩坑的时间非常久,主要是没搞懂几个东西的关系,就在瞎调试,以及当时很多东西不懂,很多报错也一知半解的。这次重装系统后正好需要再配置一次,把再一次的经历记录一下。我的电脑是华为的matebook13,intel i5-8625U,MX250显卡,win10系统。(不得不吐槽很垃圾,只能满足测试测试调调代码的需求)

深度学习利用Tensorflow平台,其中的Keras Sequential API对新用户非常的友好,可以将各基础组件组合在一起来构建模型。

(官网: https://tensorflow.google.cn/?hl=zh-cn)

Window10上Tensorflow的安装(CPU和GPU版本)

Window10上Tensorflow的安装(CPU和GPU版本)

安装Tensorflow 分为 tensorflow_cpu 和 tensorflow_gpu版本

GPU就是用来渲染计算的,GPU版本计算性能是CPU的百倍之快。如果电脑没有独立显卡只能用CPU版本计算。

Window10上Tensorflow的安装(CPU和GPU版本)

CPU版本安装:

tensorflow_cpu版本只需要安装anaconda后在anaconda prompt里面pip install tensorflow_cpu==(版本号) 即可。安装anaconda的方法见GPU版本里面。

注意查下python和tensorflow_cpu适配的版本号。

Window10上Tensorflow的安装(CPU和GPU版本)

GPU版本安装

tensorflow_gpu版本安装大致分为三步:1、安装anaconda 2、安装cuda和cdunn 3、安装tensorflow_gpu。其中第一步和第二步的顺序可以调换,就是安装完了前面三个东西再安装tensorflow即可。

tensorflow最近出了2.0版本,和1.0大版本有一些区别。具体我还没有去了解,代码不一定兼容,需要注意一下。这三个步骤的版本需要格外的注意,一旦三个自己的版本互相不兼容或者和电脑的显卡不兼容,就用不了。所以安装前看看要安装的tensorflow版本。先查好显卡的算力,然后适配的相应版本再安装,否则踩坑要很久。

首先确定电脑所能支持的tensorflow版本,根据tensorflow官网所给配置,我们要去检查电脑gpu的cuda支持版本,再去对应下载python版本和tensorflow版本。

右键 桌面>NVIDIA控制面板>帮助>系统信息>组件

Window10上Tensorflow的安装(CPU和GPU版本)

上图说明我的显卡所支持的cuda版本为11.1(向下兼容)

Tensorflow配置window官网:https://tensorflow.google.cn/install/source_windows

Window10上Tensorflow的安装(CPU和GPU版本)

以我电脑为例,之前我用的是3.7.3的版本。这次重装系统后安装的是3.8.3。

cmd查看python版本:Window10上Tensorflow的安装(CPU和GPU版本)

我之前tensorflow用的是1.13的版本,这次因为已经安装了python3.8,因此打算安装2.0版本。如果仍需要低版本需要重新安装python较低的版本或者搭一个虚拟环境(但不是很建议)。

第一步,安装anaconda

anaconda会对应安装python环境,不一定要最新的, 最新的python版本不一定有兼容的cuda加速,有cuda加速也不一定支持电脑的显卡。

官网安装:https://www.anaconda.com/products/individual,之前版本如果官网没有可能需要找资源。

第二步,安装cuda工具包

(官网:https://developer.nvidia.com/zh-cn/cuda-toolkit)

CUDA是 NVIDIA 专为图形处理单元 (GPU) 上的通用计算开发的并行计算平台和编程模型。借助 CUDA,开发者能够利用 GPU 的强大性能显著加速计算应用。在经 GPU 加速的应用中,工作负载的串行部分在 CPU 上运行,且 CPU 已针对单线程性能进行优化,而应用的计算密集型部分则以并行方式在数千个 GPU 核心上运行。使用 CUDA 时,开发者使用主流语言(如 C、C++、Fortran、Python 和 MATLAB)进行编程,并通过扩展程序以几个基本关键字的形式来表示并行性。由于tensorflow最高版本对应的是cuda10.1版本,那下载cuda10.1即可。安装包链接:https://developer.nvidia.com/cuda-toolkit-archive

Window10上Tensorflow的安装(CPU和GPU版本)

Window10上Tensorflow的安装(CPU和GPU版本)

Window10上Tensorflow的安装(CPU和GPU版本)

CUDA的下载需要挂载VPN,否则下下来只有1kb (好像后面又不一定需要VPN,可以直接下载试试)

自定义安装,安装的东西全选了,尽量不要改安装位置

Window10上Tensorflow的安装(CPU和GPU版本)

安装完之后安装cuDNN, cuDNN是用于深度神经网络的GPU加速库.注意安装的版本,选择自身cuda版本对应的cudnn下载。

官网:https://developer.nvidia.com/rdp/cudnn-archive ,cuDNN的下载需要注册官网的账号

Window10上Tensorflow的安装(CPU和GPU版本)

下载好的cudnn文件解压后,将文件夹内的文件放到cuda对应文件夹下,注意:是文件夹内的文件,而不要直接复制替换文件夹

Window10上Tensorflow的安装(CPU和GPU版本)

第三步,安装tensorflow—gpu

在anaconda prompt里面安装:pip install tensorflow-gpu , 后面可以指定版本号,下载慢挂载一个镜像源 -i https://pypi.tuna.tsinghua.edu.cn/simple

Window10上Tensorflow的安装(CPU和GPU版本)

安装完成:

Window10上Tensorflow的安装(CPU和GPU版本)

测试是否安装成功:

import tensorflow as tf
tf.test.is_gpu_available()

Window10上Tensorflow的安装(CPU和GPU版本)

Window10上Tensorflow的安装(CPU和GPU版本)

True即表示安装成功了。

查看版本号:Window10上Tensorflow的安装(CPU和GPU版本)

至此安装成功,可以跑一个小程序测试一下。

import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
 
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
print('GPU:', tf.test.is_gpu_available())

显示“GPU True”, 也即代表GPU版本安装成功。

经过几天调了一下代码之后发现tensorflow2相对与1还是有挺多改动的,1里面能够运行的代码可能2里面需要一定的修改。

关于Tensorflow2和1上面keras的一些区别可以搜一下相关的资料。

到此这篇关于Window10上Tensorflow的安装(CPU和GPU版本)的文章就介绍到这了,更多相关Window10安装Tensorflow 内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
在Windows8上的搭建Python和Django环境
Jul 03 Python
使用Python制作获取网站目录的图形化程序
May 04 Python
在Python的Django框架中创建语言文件
Jul 27 Python
简单理解Python中的装饰器
Jul 31 Python
Python数据库的连接实现方法与注意事项
Feb 27 Python
简单学习Python time模块
Apr 29 Python
Python实现导出数据生成excel报表的方法示例
Jul 12 Python
python读取视频流提取视频帧的两种方法
Oct 22 Python
Python实现的生产者、消费者问题完整实例
May 30 Python
python 矢量数据转栅格数据代码实例
Sep 30 Python
django-xadmin根据当前登录用户动态设置表单字段默认值方式
Mar 13 Python
Python系统公网私网流量监控实现流程
Nov 23 Python
详解python中的异常捕获
Dec 15 #Python
python 6种方法实现单例模式
Dec 15 #Python
Ubuntu16安装Python3.9的实现步骤
Dec 15 #Python
Python爬虫开发与项目实战
Dec 16 #Python
python中reload重载实例用法
Dec 15 #Python
python 实现socket服务端并发的四种方式
Dec 14 #Python
linux centos 7.x 安装 python3.x 替换 python2.x的过程解析
Dec 14 #Python
You might like
日本十大最佳动漫,全都是二次元的神级作品
2019/10/05 日漫
解析curl提交GET,POST,Cookie的简单方法
2013/06/29 PHP
ajax+php控制所有后台函数调用
2015/07/15 PHP
Yii2简单实现给表单添加验证码的方法
2016/07/18 PHP
php+lottery.js实现九宫格抽奖功能
2019/07/21 PHP
如何在PHP中生成随机数
2020/06/04 PHP
让mayfish支持mysqli数据库驱动的实现方法
2010/05/22 Javascript
JavaScript 操作键盘的Enter事件(键盘任何事件),兼容多浏览器
2010/10/11 Javascript
JS上传前预览图片实例
2013/03/25 Javascript
wap手机图片滑动切换特效无css3元素js脚本编写
2014/07/28 Javascript
百度地图api如何使用
2015/08/03 Javascript
js中的关联数组与普通数组详解
2016/07/27 Javascript
bootstrap table复杂操作代码
2016/11/01 Javascript
详解Vue的computed(计算属性)使用实例之TodoList
2017/08/07 Javascript
vue 自定义全局方法,在组件里面的使用介绍
2018/02/28 Javascript
详解vue挂载到dom上会发生什么
2019/01/20 Javascript
JavaScript构造函数原理及实现流程解析
2020/11/19 Javascript
[14:50]2018DOTA2亚洲邀请赛开幕式
2018/04/03 DOTA
python中bisect模块用法实例
2014/09/25 Python
Python实现的一个简单LRU cache
2014/09/26 Python
详细探究Python中的字典容器
2015/04/14 Python
Python 常用string函数详解
2016/05/30 Python
Windows上使用virtualenv搭建Python+Flask开发环境
2016/06/07 Python
python获取list下标及其值的简单方法
2016/09/12 Python
flask框架实现连接sqlite3数据库的方法分析
2018/07/16 Python
彻底理解Python中的yield关键字
2019/04/01 Python
python时间与Unix时间戳相互转换方法详解
2020/02/13 Python
解决python便携版无法直接运行py文件的问题
2020/09/01 Python
python递归函数用法详解
2020/10/26 Python
PHP开发的一般流程
2013/08/13 面试题
营销人才自我鉴定范文
2013/12/25 职场文书
某集团股份有限公司委托书样本
2014/09/24 职场文书
初中教师个人工作总结
2015/02/10 职场文书
老人院义工活动感想
2015/08/07 职场文书
Python 内置函数速查表一览
2021/06/02 Python
实操Python爬取觅知网素材图片示例
2021/11/27 Python