安装windows tensorflow-gpu

1.用驱动精灵安装nvidia最新版本驱动

2.下载cuda8.0 https://developer.nvidia.com/cuda-downloads

3.下载cudnn,需要注册登录下载 https://developer.nvidia.com/cudnn

我win7系统下载的:cudnn-8.0-windows7-x64-v5.1

解压将bin-include-lib3个目录复制粘贴到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0目录下.

环境变量,安装的时候,会自动配置好的,不需要你配置.

pip install tensorflow-gpu

此时,就可以正常运行了.

运行的时候黑屏,是因为GPU显存不够用,将一次训练的图片设置小一点就可以了.