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

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

posted @ 2017-10-20 20:20  Irran  阅读(966)  评论(0编辑  收藏  举报