Windows下安装Tensorflow GPU

Windows系统配置Tensorflow（GPU）

目录：

• 前置条件
• 安装步骤
• 常见错误

---

前置条件

1. 在CUDA GPUs官方网站中查询个人电脑中的显卡是否支持GPU
2. 已在windows系统中安装了python3.5（推荐）和pip（默认已安装）

• 打开设备管理器，例如下图中显卡型号为940MX：
  

• 在官网中点击GeForce查找是否存在该型号：
  

---

安装步骤

---

安装GPU的Tensorflow

• 打开命令行界面，输入以下命令：

pip install tensorflow-gpu

---

安装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 tfa = tf.constant("hello")b = tf.constant("world")op = a + bsess = tf.Session() sess.run(op)sess.close()

---

常见错误

• CUDA安装了8.0以外的版本
• 解决方案：
  打开控制面板的卸载界面，留下红色箭头所示NVIDIA的两个重要软件：NVIDIA图像驱动软件和NVIDIA Physx系统软件，其他的都卸载