首先要说,官网上的指南是最好的指南。
https://www.tensorflow.org/install/install_windows
需要*看。
想要安装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其实是个压缩包,先解压缩,然后有两种办法安装
- 将解压后的bin, include, lib 三个文件夹复制到 cuda安装目录的对应文件夹下。默认的情况下,cuda会安装在 C:\Program Files\NVIDIA\GPU\Computing\Toolkit\CUDA\v8.0
- 将解压得到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环境变量。
唉,当初我安装的时候就是一直出现这个问题,整整战斗了一天才成功。