1.环境配置
Tensorflow-gpu==1.3.0 + CUDA8.0 + CUDNN6.0
解释一下,现在的Tensorflow-GPU版本已经到了2.0.0a0版本,因为本人的笔记本显卡GTX-1060的显卡只能安装CUDA8.0版本,所以对应的Tensorflow-GPU版本只能选1.4.0以下的版本。
2.文件下载
CUDA8.0 https://developer.nvidia.com/cuda-downloads
CUDNN6.0 https://developer.nvidia.com/rdp/cudnn-archive
pip install tensorflow-gpu==1.3.0 -i https://pypi.douban.com/simple
文件下载之后,进行安装
3.替换CUDA8.0中的文件
安装完成之后 将下载的cudnn-8.0-windows10-x64-v6.0.zip文件解压后,
将cudnn-8.0-windows10-x64-v6.0\cuda\bin下的文件复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
将cudnn-8.0-windows10-x64-v6.0\cuda\include 下的文件复制到 C:\Program Files\NVIDIA GPU Compu