Win10下TensorFlow-GPU安装与使用

本机环境：Win10,Anaconda3,VS2015,GTX1080

Step 1：pip3 install tensorflow-gpu

Step 2:安装CUDA_8.0.61和CUDNN_v6.1,这里要注意两个版本要匹配，比如CUDA_8.0.61配CUDNN_v5.1就不行

Step 3：将解压后的CUDNN中的文件按相应的目录复制到CUDA安装的目录中：

X:\cuda\bin\cudnn64_6.dll —> Y:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin 

X:\cuda\include\cudnn.h —> Y:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include 

X:\cuda\lib\x64\cudnn.lib —> Y:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64

Step 4.设置环境变量，在Path中加上CUDNN解压后的文件所在路径

---

结尾 

尝试在手写数字集内测试，迭代次数2w,GPU版本耗时3min,在tensorflow-CPU版本下花费半个多小时。