win10系统下安装TensorFlow GPU版本

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

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环境变量。

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

转载于:https://www.cnblogs.com/irran/p/tensorflow_gpu_win10.html