Win10 Anaconda TensorFlow-GPU版本

Win10 Anaconda TensorFlow-GPU版本

笔者安装日期为2018.8.1

我的硬件设备为：

• 操作系统：win10 64位
• 显卡：NIVIDIA GeForce 940MX

安装GPU版本的TensorFlow应该先查看CUDA-Enabled GeForce Products
查看自己计算机的显卡是否满足可参考blog:
https://blog.youkuaiyun.com/u010159842/article/details/56666158/

安装过程包括以下部分：

1. 安装Visual Studio
2. 安装CUDA Toolkit
3. 下载cudnn
4. 安装Anaconda
5. 安装TensorFlow-GPU

---

安装过程

安装visual studio

下载地址：
https://visualstudio.microsoft.com/zh-hans/downloads/
我安装的community 2017版本

下载完成后正常安装即可，遇到以下选择工作负载的类似界面时，选择“Windows（3）”中的“使用C++的桌面开发”，确认后继续接下来的步骤，完成安装即可。

后续，如果还需要添加其他工作负载时，可以找到visual studio installers

双击打开，并选择更多—>修改，即可进入选择工作负载的界面

安装CUDA Toolkit
(我安装的为CUDA Toolkit 9.2)
下载地址：
https://developer.nvidia.com/cuda-release-candidate-download

CUDA toolkit 9.2版本提供了local版本和network版本；注意下载exe(local)，大概1.5G； exe(network)的选项，大概15M

安装时可以选择精简安装和自定义安装，自定义安装可以选择安装哪些驱动程序组件、安装位置等。

验证CUDA是否安装完毕：
打开cmd，输入nvcc –V，显示如下图则表示安装完毕。

其他更多验证方法可以参照blog：
https://blog.youkuaiyun.com/m0_37638031/article/details/78896818

下载cudnn
下载地址：
https://developer.nvidia.com/rdp/cudnn-download

下载与CUDA版本对应的cuDNN，对应CUDA9.2我得到一个安装包cudnn-9.2-windows10-x64-v7.1，解压之后将其中如下图所示的3个文件夹拷贝至CUDA9.2的安装路径根文件夹下：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2

（此处需要根据自己的安装情况修改路径，安装了9.0版本则为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0）

安装Anaconda
（Anaconda的下载及仓库镜像配置我是参照雪晴数据网来进行的）
雪晴数据网相关网页：
http://www.xueqing.tv/course/90

下载好anaconda套件，之后需进行的步骤是：

（1）进行安装
视频：http://www.xueqing.tv/lesson/342
安装遇到以下界面时，注意勾选“Add Anaconda to my PATH environment variable”

没有勾选的话需要自己添加系统变量
（按下Win+R，运行control打开控制面板，选择系统—>高级系统设置—>环境变量，即可查看和修改系统变量）

（2）配置anaconda仓库镜像
视频：http://www.xueqing.tv/lesson/348

网页上其余的下载以及教学步骤可以不管。

 
安装TensorFlow
（1）打开Anaconda Prompt

（2）输入指令：
conda install tensorflow-gpu

验证是否安装完毕：
在命令行中，进入python，并输入以下代码，或者打开Spyder在输入以下代码。

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

输出为：

b’ Hello, TensorFlow!