Win10环境+ CUDA9.0 +CUDNN7.0+TensorFlow1.7/1.6/1.5配置

前言：很多小伙伴在选择CUDA版本和cudnn版本上有疑问，这里简短的说一下，希望能帮到各位小伙伴，我在网上看到有人说，要根据自己的显卡来选择CUDA，其实是错误的，你可以下载自己所需要的CUDA版本,这只是个驱动而已。CUDA的版本和cundd的版本要对应，不然可能会有不兼容报错。CUDA是显卡驱动程序，cundd是用来加速深度学习训练的库，在训练网络时，使用GPU加速，会调CUDA驱动和cudnn库，安装时两者要联系起来，本文的第二步，就是做这个事。

系统环境：windows10     64位

显卡：GTX1060

CUDA版本:cuda_9.0.176_win10（配套的cudnn一定要是7版本）

下载地址：https://developer.nvidia.com/cuda-90-download-archive

CUDNN版本:cudnn-9.0-windows10-x64-v7（7版本的）

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

TensorFlow-gpu版本：TensorFlow1.7（1.6和1.5也支持）

1.下载CUDA, https://developer.nvidia.com/cuda-90-download-archive

按照自己机器的信息选择，下图是我自己机器信息：

下载完之后直接点击运行，会自行安装，安装时会让你选择一个文件夹作为抽取文件临时存放，记住这不是安装文件夹，安装文件夹系统会自动安装到C盘，具体我后面会提到，临时文件夹安装完后会自动删除。

2.下载cudnn,下载地址：https://developer.nvidia.com/cudnn，进去之后,找到对应的版本，点击Download，他会要你注册登录，然后填个问卷，才可以下载，这一步自行按照网站提示做就可以了。提交完问卷后，就可以点击下载了。以下是版本选择界面：

勾选这个，然后选择要下载的版本，我选的是图中划线那个（应该选择7.0版本，7.1可能有些不兼容）。

下载完，解压cudnn,会出现四个文件，如下：

接下来，我们要将CUDA和CUDNN链接起来，这样CUDA才能调用CUDNN.这里有两种方法将CUDA和CUDNN链接起来

第一种：

在环境变量下面，把解压后文件的bin目录加到系统环境变量中，放在path下面，如图：

重启一下电脑，让环境变量生效，完成。

第二种：

就是把cudnn中的库加入到CUDA中，因为CUDA系统会自动安装到c盘的Program Files文件中，选择该文件夹下的NVIDIA GPUComputing Toolkit文件，一直到

这个文件夹下有如下文件，我标出的这三个，和cudnn中的前三个文件是对应的如下图：

接下来要做的就是，把cudnn三个文件夹中的文件，复制到CUDA相应文件夹的下面（仔细看一下就会发现他们的目录结构是一样的），完成。

3.安装TensorFlow

安装：在命令窗口中输入pip install tensorflow-gpu，

如果要安装（更新）最新的版本，可输入： pip install --upgrade tensorFlow-gpu

等待安装完成即可

4.现在CUDA,cudnn,TensorFlow都已经安装好了，可以尽情的去写TensorFlow代码啦！

（注：新手第一次使用TensorFlow-GPU版时，训练网络时应该在tf.session处做一点修改，如下：

将with tf.Session() as sess:

改为：

config = tf.ConfigProto()
config.gpu_options.allow_growth = True

with tf.Session(config=config) as sess:

否则可能会报错：

原因是当TensorFlow要大量使用GPU时，不会自动给其分配资源，需要手动调整。

）