win10下安装tensorflow-gpu(python3.5.4)
前提: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安装完毕