Win10下安装Tensorflow GPU版本记录

1. 前言

这里我使用的环境是Win10家庭中文版，显卡是GTX1050。由于安装Tensorflow相对安装Caffe来说比较简单，这里就直接给出安装步骤了。

1. 安装CUDA8

安装CUDA需要到这里(https://developer.nvidia.com/cuda-toolkit-archive)去下载，但是呢！需要注意一点的是CUDA的版本和cuDNN的版本要对应上，要不是会出问题。这里我选择的是CUDA8，下载之后就是安装了安装和一般的exe没啥区别，这里就过…

2. 安装cuDNN6

cuDNN是专用于神经网络计算加速的，其实他就是一包库文件，每一个CUDA版本都有特定与之匹配的cuDNN库文件，如CUDA 8.0就匹配cudnn-8.0-windows10-x64-v6，通常我们到官网上下载并解压缩后，直接将其文件替换到CUDA相应目录下即可！如下，解压缩cuDNN后在cuda目录下得到三个文件夹，里面分别有cudnn64_6.dll、cudnn.h及cudnn.lib。

我们将其拷贝合并到CUDA的安装目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0下

之后便是设置环境变量了，环境变量设置如下：

3. 安装Tensorflow

这里看了很多资料说要装什么Anaconda，这里就直接pip3了。不整那么麻烦了。

pip3 install tensorflow-gpu==1.4

这里指定了版本是因为当前安装的CUDA与cuDNN只支持哪个版本的，所以只能安装这个版本的了。若是需要安装1.5及其以上的版本就需要安装CUDA9与对应的cuDNN了。
接下来就可以开始使用Tensorflow了。