环境: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版本。
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的显卡之后就一切正常了。