Windows10上安装tensorflow-gpu流程

前言

---

最近在windows10上安装了TensorFlow(gpu版)，网上也有很多安装教程，但是在实际安装中还是遇到了许多问题，在这里把自己的安装步骤写下来作为一个参考和补充。

---

安装材料：

---

1.CUDA® Toolkit 8.0.
2.cuDNN v6.0
3.Visual Studio 2015(Community)

百度云连接：TensorFlow(gpu)安装程序
PS：这里默认GPU在CUDA中的计算能力高于3，详见 NVIDIA documentation

---

安装过程

---

1.Visual Studio 2015(Community)

首先安装的是 Visual Studio 2015(Community)，安装步骤这里就不在赘述了，有需要的同学可以参考：Visual Studio 2015(Community)安装教程

2.CUDA 8.0

百度云提供的版本是cuda_8.0.61_win10，这是大部分人安装的版本，如果读者想自己下载CUDA，请戳这里NVIDA CUDA(有的下载链接打不开，我的就是在其他地方找的)

安装完成后，测试CUDA8，打开cmd，输入”nvcc -V”，得到下图，即为安装成功。

接下来要运行CUDA的样例程序，打开C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0目录下的Samples_vs2015.sln

将这里调成Release和X64模式

然后右键解决方案，点击”生成解决方案”，然后耐心等待。

编译成功后，使用cmd进入c:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64\Release目录下，然后运行deviceQuery.exe。显示下图则表示CUDA成功。

PS：这里对CUDA的安装方式，参考了TF-Windows10安装TensorFlow-GPU版的博客，也有一些教程，在CUDA的安装后，并没有使用VS 2015 调整模式，而是选择配置环境变量，读者可以自行参考。

3.安装cuDnn6.0
百度云提供的是cudnn-8.0-windows10-x64-v6的版本，如果读者想自行选择版本下载的话，请戳这里：cuDnn下载(注意下载的版本一定要跟CUDA8.0是配套的，像这种是下了也没有用的。)

下完之后，解压缩会发现里面有三个文件夹，分别是bin、include和lib，把他们复制到CUDA的安装路径就可以了。CUDA默认安装路径为：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0。

敲黑板！！敲黑板！！敲黑板！！
这里会有一些由于TensorFlow版本不同造成的问题，请注意！
如果要安装的TensorFlow是1.3版本的话，那么要选择cuDnn6.0！！！
如果要安装的TensorFlow是1.0版本的话，那么要选择cuDnn5.1！！！
(我当时不信邪，使用的cuDnn7.0版本，然后改cudnn64_7.dll的名字，结果在CNN的时候就报错内存溢出了，重新使用cuDnn6.0的版本才可以使用。)

4.安装TensorFlow
现在python3.6也提供TensorFlow(gpu)版本的安装了，所以不需要强制用3.5版本安装TensorFlow。直接输入

pip3 install --upgrade tensorflow-gpu

如果有PyCharm编译器也可以使用PyCharm安装。当然，有的人也会使用Anaconda安装，这里就不再赘述了。

---

测试TensorFlow

---

可以在编译器中输入如下代码：

import tensorflow as tf
# 新建一个graph.
with tf.device('/gpu:0'):
#选择gpu或cpu运行
  a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
  b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
# 新建session with log_device_placement并设置为True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# 运行这个op.
print(sess.run(c))

如果出现下图，则表示安装成功。