之前在实习公司的电脑上装过TensorFlow-gpu,那时候很快就装好了。但在自己的笔记本上装时,却搞了很久。。。
一部分原因是因为用校园网下载cuda toolkit 和cudnn ,总是在最后时刻罢工。辗转叫了几个人帮忙,才把这两个东西下载好。。
安装步骤
其实,这个安装过程还是蛮简单的。据说cuda版本和cudnn版本没选好的话,就会遇到很多坑,所幸,我选的版本都能很好的驾驭tensorflow。
下面是要安装的东西,可以依次安装,也可以按你喜欢的顺序来安装。但要保证安装了VC++插件和python之后,才安装tensorflow。
Visual C++ Redistributable for Visual Studio 2015: tensorflow其实是基于VC++2015开发的,所以需要安装Visual C++ Redistributable for Visual Studio 2015。下载后,直接默认安装就行
python环境:我用的是anaconda3.x,新建了一个python3.5的环境,然后在python3.5中安装的tensorflow-gpu。这个不会的话,可以尝试搜索“anaconda安装、anaconda清华源安装”和“anaconda新建python环境”。。
cuda9.0: cuda-v9.0.下载之后,默认安装就行,中间要选同意的地方就选同意。
安装好cuda后,要配置系统环境。用小娜搜索“控制面板”–>选择大图标–>选择系统–>选择高级系统设置–>选择环境变量–>选择path,新增加
“C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0”
–>确定。。然后再新增几个变量,如下图(已经存在的就不用增加了):
cudnn:(cuDNN v7.1.4 Library for Windows 10) 下载好后,解压,把所有东西复制到
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0
中,就ok啦tensorflow-gpu:安装win+R,运行cmd,打开命令窗口,激活python3.5的环境(activate 环境名),然后输入
pip install tensorflow-gpu
即可。等待安装成功就行。
测试
我喜欢用jupyter notebook来写python,所以现在也用它来写一下测试代码:
import tensorflow as tf
hi = tf.constant('Hi , welcome to TensorFlow ! ')
sess = tf.Session()
print(sess.run(hi).decode())
得到结果如下:
cool,顺利输出了结果,说明安装成功。