window10系统下python3.5.0安装tensorflow-gpu

时间:2024-05-30 19:12:17

 

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后默认直接安装,系统变量会出现几个路径

window10系统下python3.5.0安装tensorflow-gpu

系统变量path路径也会多出2个路径:

window10系统下python3.5.0安装tensorflow-gpu

cmd下输入nvcc -V查看是否安装成功

window10系统下python3.5.0安装tensorflow-gpu

 

下载cudnn7版本文件

文件复制到对应的文件名的文件中

 

然后安装pip install tensorflow-gpu==1.10.0

window10系统下python3.5.0安装tensorflow-gpu

没有错误提示安装成功

 

检查是否可以正常调用gpu跑程序:

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

window10系统下python3.5.0安装tensorflow-gpu

出现这两个名字就表示tensorflow-gpu版本的已经安装成功了