python版本:python3.5.0
显卡版本:gtx 780 (计算能力3.5)
系统:win10 64位
cuda版本:cuda9
cudnn版本:cudnn7版本
tensorflow-gpu版本:tensorflow-gpu1.10.0
vs版本:安装visual studio 2015(旗舰版)
注意事项:
1:cuda,cudnn,tensorflow,python,这4个版本一定要全部对应,防止出错浪费时间
2:tensorflow-gpu1.10.0需要gpu计算能力3.5以上,大于tensorflow1.11.0的就需要计算能力3.7以上(血的教训,我的刚好3.5)。
3:安装cuda后要配置系统环境变量,cudnn内容要复制到cuda对应的文件夹中
4:最好检查用驱动精灵,360驱动更新显卡驱动到最新版本
步骤
一:安装python3.5.0版本,并配置环境变量。更新pip和setuptools两个工具包
下载cuda9.0后默认直接安装,系统变量会出现几个路径
系统变量path路径也会多出2个路径:
cmd下输入nvcc -V查看是否安装成功
下载cudnn7版本文件
文件复制到对应的文件名的文件中
然后安装pip install tensorflow-gpu==1.10.0
没有错误提示安装成功
检查是否可以正常调用gpu跑程序:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
出现这两个名字就表示tensorflow-gpu版本的已经安装成功了