安装TensorFlow-GPU版本

环境：win10+GTX1060+Python 3.6.8

安装：TensorFlow 1.14.0+CUDA 10.0+cuDNN 7.6.0

第一步：安装TensorFlow

pip install tensorflow-gpu

第二步：下载CUDA和cuDNN

在Python安装路径下进入tensorflow\python\platform打开build_info.py

该文件显示了已经安装好TensorFlow匹配的CUDA和cuDNN版本。

CUDA下载地址：https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

cuDNN下载地址：https://developer.nvidia.com/rdp/cudnn-archive#a-collapse705-9

有很多匹配版本，我没选择最新版，选择了cuDNN 7.6.0版本。

第三步：安装

1.CUDA安装。

我是VS2019，所以安装CUDA会显示无法检测到vs，提示先行安装vs。我选择了忽略。

2.cuDNN是直接解压到文件夹。

解压后如下图：

将cuDNN的三个文件夹拷贝到CUDA的安装路径中，文件夹进行合并即可：

第四步：修改环境变量

修改系统变量path，增加两项：

第五步：检查gpu版本是否安装成功，测试代码：

import tensorflow as tf
print(tf.test.is_built_with_cuda())

返回True即安装好了。

第六步：中间踩了一个坑，在安装CUDA和cuDNN过程中，显卡GTX1060的驱动莫名奇妙坏掉了，集显驱动能用，导致环境变量配好以后，测试代码也检测不到GPU。更新了1060的显卡之后就一切正常了。