如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走

看到有很多朋友问TensorFlow-gpu如何配置,我打算分享下我的安装经验。

first and first

兄dei,你的显卡是NVIDIA 1060 以上吗?
不是的话,先去买个深度学习 穷人入门版 的1060的显卡吧(最低1050ti才可以在深度学习领域划水),在公司工作的、土豪建议直接2080Ti,或者专业图形显卡P100

只有NVIDIA显卡支持GPU加速

TITAN系列显卡

如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第1张图片

专业图形显卡

如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第2张图片

首先看看你的系统配置,WIN10系统上的鼠标右键打开NVIDIA控制面板,
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第3张图片
找到“组件”里的CUDA型号,比如我的就CUDA10.2
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第4张图片

安装TensorFlow-gpu

到如下网址查看你的匹配的TensorFlow-gpu的版本
https://www.tensorflow.org/install/source_windows
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第5张图片
在你的虚拟环境里(最好用anaconda创建python=3.6虚拟环境,python3.7太新新,有些git代码不支持,python3.5太老,不知道如何创建anaconda虚拟环境的百度),输入如下命令,记得选择对应的TensorFlow-gpu版本安装

pip install tensorflow-gpu==1.14 -i http://pypi.douban.com/simple

如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第6张图片

准备CUDA和cuDNN安装文件

打开下面的网址下载,符合你电脑的CUDA
https://developer.nvidia.com/cuda-toolkit-archive
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第7张图片
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第8张图片

进入一下网站https://developer.nvidia.com/rdp/cudnn-download 下载相配的cuDnn 版本
进这个网要注册登录一下,
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第9张图片
才能看到如下界面,图中有写明cuDNN 几点几对应 CUDA哪个版本。注意选取下载。
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第10张图片
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第11张图片
现在有了CUDA和cuDNN两个必备的文件,接下来先双击第一个文件cuda.exe进行安装
在这里插入图片描述

CUDA安装

安装路径可以默认也可以自定义。之后点击ok就行
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第12张图片
等一会,加载。。。
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第13张图片
keep waiting(好事多磨,再等会。。。)
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第14张图片
点同意意,继续
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第15张图片
如果不知道怎么选,就选择精简安装。
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第16张图片
选默认或自定义,记下安装路径
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第17张图片
最终“下一步”,然后“完成”就行。
配置系统环境变量,选择path:
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第18张图片
以下两个是自动就有的,没有自己加一下。
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第19张图片
win+R打开cmd,验证安装:nvcc -V
出现如下图代码,就是成功安装CUDA了
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第20张图片

配置cuDNN

接下来,该用到第二个文件
在这里插入图片描述
解压后出现bin、include、lib三个文件夹
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第21张图片
一个一个复制粘贴到刚才的CUDA安装路径下
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第22张图片
接下来添加cudnn的path
这里特别重要,改成你自己的路径

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第23张图片

至此cuDNN安装完成!!

测试

1、进入cmd激活你装好TensorFlow-gpu环境,键入python进入python shell;
2、输入import tensorflow as tf导入tensorflow库,无报错即成功安装TensorFlow;
3、键入代码

hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()

有如下图的效果,即说明使用了GPU进行计算。
如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走_第24张图片

提醒

pycharm里运行TensorFlow-gpu会报一些warning,那些提醒是因为版本太新或者有些稍微不匹配的问题,不过不用担心,只要会出结果,能运行就是成功了。

tensorflow-gpu安装成功!

好用点个赞呗!本人持续更新python人工智能方面的东西。大家一起进步!

你可能感兴趣的:(AI)