TensorFlow的gpu版本在windows下安装指南

当前环境描述：Win10 64位，Python3.6

目标：安装使用TensorFlow1.6-GPU

所需文件：Cuda9.0、Cudnn7.0

注意：安装Cuda9.1 & Cudnn7.1的环境下将不能正常使用TF1.6-GPU

1、下载安装Cuda9.0

网址：

https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10

注意：搜索CUDA进入下载界面默认下载最新的9.1版本，不要下这个版本

步骤：搜索cuda9，进入后点击下载，下载完成直接安装即可

安装完成后 在shell窗口下，输入命令nvcc -V

即可查看版本信息

2、下载安装Cudnn7.0、添加进path

网址：https://developer.nvidia.com/rdp/cudnn-download

注意：下载Cudnn会需要注册，并且完成问卷调查

步骤：下载完成后为安装包形式，解压后，将文件夹下面的bin目录添加到系统path变量里面

我的cudnn的bin目录：C:\Program Files\cudnn-9.0-windows10-x64-v7.1\cuda\bin

3、安装tensorFlow

在windows power shell下，输入命令

pip install --ignore-installed --upgrade tensorflow-gpu==1.6.0

即可安装1.6.0版本的tensorflow，注意1.6版本的tensorflow必须安装9.0的cuda以及对应的cudnn，别的版本需要看tensorflow给的报错提示，一般是会提示你的，我开始按得8.0的cuda就提示我出错，叫我去安装9.0的cuda

4、测试

进入python的交互式界面

依次输入

import tensorflow as tf

a = tf.random_normal((100, 100))
b = tf.random_normal((100, 500))
c = tf.matmul(a, b)
sess = tf.InteractiveSession()
sess.run(c)

如果出现这样的界面就说明tensorflow运行正常，安装已经完成了

注：我使用了anaconda创建了了一个python3.5的子环境去安装tensorflow的，这个子环境里面没有办法使用anaconda自带的那些库，这是我们需要在子环境里面再安装一个anaconda，可以使用下面的命令：

conda install anaconda

这样可以在子环境里面使用它丰富的类库了。