windows10+CUDA8+cuDNN6 tensorflow（GPU版）安装经历

环境说明

• 系统：windows10 家庭版
• GPU型号：NVIDIA Geforce GT 755M
• Python版本：Anaconda 5.01 （Python 3.6 version）

---

安装GPU版的要求

• CUDA® Toolkit 8.0
• cuDNN v6.1

• GPU的CUDA计算能力在3.0以上

  在这个网站上可以查看GPU是否支持CUDA以及GPU对应的计算能力：https://developer.nvidia.com/cuda-gpus

---

CUDA安装

这里是8.0版本的下载地址，现在官方要求的也是安装8.0版本的CUDA：https://developer.nvidia.com/cuda-80-ga2-download-archive
选一下系统版本下载下来安装就行了。这一个地方自动更新了我的显卡驱动。原本我是较老的328版本，现在变成了376.51版本。

cuDNN

现在官网上最新的版本是7.0，可以直接在那个界面点击download，注册之后就可以看到历史版本。Tensorflow的官方安装指南里写的是安装cuDNN 6.1，不过我在历史版本里只看见了cuDNN 6 for CUDA 8，就选择了这个版本下载安装。
下载下来之后，解压压缩包，把其中的所有内容，包括bin、include、X64都拷贝到CUDA的安装目录中。默认安装的话，位置是在：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

tensorflow安装

在命令行中输入

pip install --upgrade tensorflow-gpu

等待一会儿，就安装成功了。

---

测试

使用一个简单的程序测试一下：

import tensorflow as tf

sess = tf.Session()
a = tf.constant(2)
b = tf.constant(3)
print(sess.run(a+b))

这一步会显示自己的显卡信息，如果没有显示错误信息，就表示安装已经成功了。

---

参考资料

1. Tensorlow官方安装教程：https://www.tensorflow.org/install/install_windows
2. 【TensorFlow】Windows10 64 位下安装 TensorFlow - 官方原生支持：http://blog.csdn.net/u010099080/article/details/53418159
3. win10下配置Tensorflow1.4：http://blog.csdn.net/Gavin__Zhou/article/details/78484004