小心大坑!大坑！win10系统下安装TensorFlow GPU版本

最近倒腾深度学习做图像立体匹配，于是着手下载tensorflow，这个东西装起来到处都是坑啊！CSDN里参考博客特别多，但是都没说的很明白，比如Windows10 必须用那个版本的cuda等。还真不要小看这些细节，不在乎的话，按照步骤 安装最后肯定是悲剧了！

在经理一次次是被之后，终于找到“对症”的良药了，cnblog一篇问把我救出坑！废话不多说上原文

【以下为转载内容】

—————————————————————————————————————————————————————————————————————————————

首先要说，官网上的指南是最好的指南。

https://www.tensorflow.org/install/install_windows

需要FQ看。

---

想要安装gpu版本的TensorFlow。我们需要安装 Cuda 和Cudnn

需要注意的是，他们的版本极其重要

cuda必须是8.0的，不能是最新版的9.0

cudnn必须是v6.0，不能使v5.1或v7.0

TensorFlow需要是1.3版本的

 

cuda可以从官网下载

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

同理，cudnn也可以从官网下载

https://developer.nvidia.com/cudnn

---

 

cuda下载完成后直接安装就行了

安装好后，确认一下是否安装成功，在命令行里输入

nvcc --version

日过出现这样类似的信息，就说明安装成功了

---

 

cudnn其实是个压缩包，先解压缩，然后有两种办法安装

1. 将解压后的bin, include, lib 三个文件夹复制到 cuda安装目录的对应文件夹下。默认的情况下，cuda会安装在 C:\Program Files\NVIDIA\GPU\Computing\Toolkit\CUDA\v8.0
2. 将解压得到bin文件夹添加到path环境变量中

---

 

然后就是安装TensorFlow了

可以直接用pip命令安装

pip install -upgrade --ignore-installed tensorflow-gpu

也可以用anaconda构建一个虚拟环境，再用上面写的pip命令安装

anaconda可以从官网下载

https://www.anaconda.com/download/

安装好后，进入命令行，创建虚拟环境

conda create -n tensorflow python=3.5

激活虚拟环境

activate tensorflow

退出虚拟环境

deactivate

---

最后测试一下TensorFlow的安装情况，在python中输入命令

import tensorflow

如果没有报错，就没问题了。

---

常见错误 No module named '_pywrap_tensorflow_internal'

如果出现这个问题，请第一时间检查cuda cudnn的版本是否正确，然后检查cudnn是否按我说的复制过去了，或者加入了path环境变量。

唉，当初我安装的时候就是一直出现这个问题，整整战斗了一天才成功。

【转载结束】

—————————————————————————————————————————————————————————————————————————————

原文连接：win10系统下安装TensorFlow GPU版本