Win10 x64下安装TensorFlow-GPU版本

安装环境：

系统：win10 64位

显卡：Geforce 1060（3G）

 

安装软件：

python 3.6 下载

vs2015运行时库：下载64位的，点击这里，下载后安装

（ps: vs run library 不一定需要，因为在安装python时会同时安装vs2015 run library）

CUDA : CUDA8_win10 版本，点击这里

cuDNN : cuDNN6.0_win10 版本

 

安装过程

1 安装python 

（ps:记得把路径添加到系统变量的path里面）

2 安装tensorflow(pip3方式) 
pip3 install --upgrade tensorflow-gpu

(ps:默认高版本的tensorflow会报错，经过试验，tensorflow-gpu==1.4.0 无错)

（ps:安装过程可能需要升级pip）

3 安装CUDA8

（1）下载安装文件，2个（有一个补丁文件）

（2）点击大的那个文件运行

（ps:我在安装过程中出现黑屏，然后就没有然后了。。。查资料后找到一个方法，卸载掉原先的显卡驱动就行。所以，强制关机，卸载驱动，再次安装，成功！）

（ps:根据别人的安装记录，此处可以选择不安装cuda驱动，应该可以避免该问题，借图如下）

（ps:安装到最后，如果本机没有安装VS2015或者VS2013、VS2012，会报如下提示，借图：）

        根据别人安装经验，在后面完成cuDNN的安装后，对tensorflow进行测试，发现是可以成功执行测试程序的。故目前尚不清楚安装Visual Studio的作用，先留白一下。 )

安装完成，验证：

4 安装cuDNN

（1）下载地址，点击这里

（2）根据之前安装的CUDA8选择下载了cuDNN v6.0

（3）下载的cuDNN是一个zip压缩文件:cudnn-8.0-windows10-x64-v6.0.zip。解压得到三个子文件夹bin、include、lib，将其拷贝到刚才安装的CUDA8的主目录，与其进行合并。

 

5 测试tensorflow

打开powershell，输入python，进入python REPL界面。输入代码测试：

 

最后总结：

（1）安装过程其实很简单，一步一步来就好

（2）报错的原因很大部分是因为版本问题，python，tensorflow, cuda, cudnn都要匹配（本文配置如下：python3.6.5, tensorflow-gpu  1.4.0, cuda8, cudnn6）

（3）接下来，准备在Linux环境下配个最新版本的环境出来

（4）补充一下：Anaconda并不是必需，可以使用可以不使用