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

first and first

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

只有NVIDIA显卡支持GPU加速

TITAN系列显卡

tensorflow显卡排行 tensorflow 显卡_python

专业图形显卡

tensorflow显卡排行 tensorflow 显卡_CUDA_02

首先看看你的系统配置,WIN10系统上的鼠标右键打开NVIDIA控制面板,

tensorflow显卡排行 tensorflow 显卡_CUDA_03


找到“组件”里的CUDA型号,比如我的就CUDA10.2

tensorflow显卡排行 tensorflow 显卡_tensorflow显卡排行_04

安装TensorFlow-gpu

到如下网址查看你的匹配的TensorFlow-gpu的版本

https://www.tensorflow.org/install/source_windows

tensorflow显卡排行 tensorflow 显卡_python_05


在你的虚拟环境里(最好用anaconda创建pythnotallow=3.6虚拟环境,python3.7太新新,有些git代码不支持,python3.5太老,不知道如何创建anaconda虚拟环境的百度),输入如下命令,记得选择对应的TensorFlow-gpu版本安装

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

tensorflow显卡排行 tensorflow 显卡_tensorflow_06

准备CUDA和cuDNN安装文件

打开下面的网址下载,符合你电脑的CUDA

https://developer.nvidia.com/cuda-toolkit-archive

tensorflow显卡排行 tensorflow 显卡_tensorflow_07


tensorflow显卡排行 tensorflow 显卡_tensorflow_08

进入一下网站https://developer.nvidia.com/rdp/cudnn-download 下载相配的cuDnn 版本

进这个网要注册登录一下,

tensorflow显卡排行 tensorflow 显卡_tensorflow_09


才能看到如下界面,图中有写明cuDNN 几点几对应 CUDA哪个版本。注意选取下载。

tensorflow显卡排行 tensorflow 显卡_tensorflow_10


tensorflow显卡排行 tensorflow 显卡_CUDA_11


现在有了CUDA和cuDNN两个必备的文件,接下来先双击第一个文件cuda.exe进行安装

tensorflow显卡排行 tensorflow 显卡_CUDA_12

CUDA安装

安装路径可以默认也可以自定义。之后点击ok就行

tensorflow显卡排行 tensorflow 显卡_CUDA_13


等一会,加载。。。

tensorflow显卡排行 tensorflow 显卡_tensorflow显卡排行_14


keep waiting(好事多磨,再等会。。。)

tensorflow显卡排行 tensorflow 显卡_tensorflow显卡排行_15


点同意意,继续

tensorflow显卡排行 tensorflow 显卡_CUDA_16


如果不知道怎么选,就选择精简安装。

tensorflow显卡排行 tensorflow 显卡_CUDA_17


选默认或自定义,记下安装路径

tensorflow显卡排行 tensorflow 显卡_tensorflow_18


最终“下一步”,然后“完成”就行。

配置系统环境变量,选择path:

tensorflow显卡排行 tensorflow 显卡_tensorflow_19


以下两个是自动就有的,没有自己加一下。

tensorflow显卡排行 tensorflow 显卡_python_20


win+R打开cmd,验证安装:nvcc -V

出现如下图代码,就是成功安装CUDA了

tensorflow显卡排行 tensorflow 显卡_python_21

配置cuDNN

接下来,该用到第二个文件

tensorflow显卡排行 tensorflow 显卡_tensorflow显卡排行_22


解压后出现bin、include、lib三个文件夹

tensorflow显卡排行 tensorflow 显卡_python_23


一个一个复制粘贴到刚才的CUDA安装路径下

tensorflow显卡排行 tensorflow 显卡_CUDA_24


接下来添加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

tensorflow显卡排行 tensorflow 显卡_tensorflow_25

至此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进行计算。

tensorflow显卡排行 tensorflow 显卡_tensorflow显卡排行_26

提醒

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

tensorflow-gpu安装成功!