win10 + cuda(v9.0) 安装TensorFlow-gpu版

之前在实习公司的电脑上装过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 Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0”–>确定。。然后再新增几个变量,如下图(已经存在的就不用增加了):
    这里写图片描述

  • cudnn:cuDNN v7.1.4 Library for Windows 10) 下载好后,解压,把所有东西复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.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,顺利输出了结果,说明安装成功。
这里写图片描述

猜你喜欢

转载自blog.csdn.net/yunyunyx/article/details/81489050