win10下安装tensorflow-gpu(python3.5.4)

来源:互联网 发布:逆战冰神鼠标宏数据 编辑:程序博客网 时间:2024/06/05 14:13

前提:dos命令下pip和python正常运行

不是内部命令或找不到程序请确认:

1.是否在path中添加了系统变量

2.用户名是否为英文


用了一下午把tensorflow的坑踩遍了!

一、首先,安装cuda:

https://developer.nvidia.com/cuda-downloads

选择,“windows”--“x86_64”--“10”--“exe(local)”,点击“Download”




二、安装cudnn:

去nvidia官网去下载cudnn:

https://developer.nvidia.com/rdp/cudnn-download

(如果网站还在维护的话,那三行文字说明这里依旧是下载不了)

所以只能这样:

链接:http://pan.baidu.com/s/1i4JQQo9

密码:pmto

(cudnn-8.0-windows10-x64-v6.0)

三、修改环境变量:左击桌面的“计算机”图标,依次选择“高级系统设置”--“环境变量”,在“系统变量(S)”中找到Path选项,

编辑,查看以下四个变量是否在其中:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\libnvvp

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include

没有就新建进去啦


四、安装tensorflow-gpu

开始菜单--windows power shell(右键以管理员权限运行)--输入"pip install tensorflow-gpu"

(不是以管理员权限运行后面安装的时候就会出一大段错误)

安装完毕。


五、检测是否安装成功

windows power shell:输入"python"

----输入"import tensorflow as tf"(没反应才说明正常)

----输入“sess=tf.Session()”

----输入“print(sess.run(tf.constant(123)))”

结果是123,ok,tensorflow-gpu安装完毕

原创粉丝点击