Windows下安装Tensorflow GPU 1.4.0
目录:
- 前置条件
- 安装步骤
- 常见错误
前置条件
- 在CUDA GPUs官方网站中查询个人电脑中的显卡是否支持GPU
- 已在windows系统中安装了python3.5(推荐)和pip(默认已安装)
打开设备管理器,例如下图中显卡型号为940MX:
在官网中点击GeForce查找是否存在该型号:
安装步骤
安装GPU的Tensorflow
- 打开命令行界面,输入以下命令:
pip install tensorflow-gpu==1.4.0
安装CUDA 8.0
打开CUDA 8.0官网,依次点击Windows->x86_64->10->exe[local],然后点击下载
下载完毕后,安装一直默认下一步即可(记住安装路径,后面步骤要用到)
(默认安装于C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0)检查系统Path变量是否添加了CUDA 8.0的路径,如果没有自动添加,手动添加其路径
安装cuDNN 6.0
进入下载界面,注册账户,填写各种繁琐的问卷调查即可安装
选择cuDNN 6.0 for CUDA 8.0,如下图所示,进行下载
解压cuDNN压缩包,会得到下图所示的文件夹
将三个压缩包选中复制到CUDA 8.0安装路径中
测试是否成功
import tensorflow as tf
a = tf.constant("hello")
b = tf.constant("world")
op = a + b
sess = tf.Session()
sess.run(op)
sess.close()
常见错误
- CUDA安装了8.0以外的版本
- 解决方案:
打开控制面板的卸载界面,留下红色箭头所示NVIDIA的两个重要软件:NVIDIA图像驱动软件和NVIDIA Physx系统软件,其他的都卸载