在Windows上安装GPU版Tensorflow

1. 下载安装Anaconda

简单说就是下载 64位 python 3.5 版本的Anaconda

https://www.continuum.io/downloads#windows

安装情况:新机,未装python。

注意

a. Windows只支持64位 python 3.5

https://www.tensorflow.org/versions/master/get_started/os_setup#pip_installation_on_windows

b. 如果import tensorflow as tf 时有errorNo module named "_pywrap_tensorflow" 或者DLL load failed, checkMSVCP140.DLLis in%PATH%。解决方法:install theVisual C++ 2015 redistributable(x64 version). 不过我没遇到这个情况。

c. Anaconda虚拟环境内外只能存在一个tensorflow,所以多出来的要删除。因为Anaconda searches system site-packages from.localwith higher priority.

2. 进入Anaconda 菜单栏下的“Anaconda Prompt”

2.1 更新pip

输入命令 python -m pip install --upgrade pip

2.2 安装Tensorflow

按照不同需求输入相应的 命令, 记得删去 "C:\>"

CPU-only version

pip install --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-0.12.1-cp35-cp35m-win_amd64.whl --ignore-installed

非GPU版在这里就可以结束了。在Anaconda Prompt输入python 后就能 导入tensorflow 包了。

我们着重讲讲GPU版!

pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl --ignore-installed

安装完后可以看到下面几个包

Installing collected packages: appdirs, six, pyparsing, packaging, setuptools, protobuf, numpy, wheel, tensorflow-gpu

安装成功后查看版本

python -c "import tensorflow; print(tensorflow.__version__)"

然后你会发现我们找到不到各种lib,不急,因为我们还没装呢。

https://github.com/ContinuumIO/anaconda-issues/issues/542)!!!

安装cuDNN和CUDA Toolkit

(win + R 后输入dxdiag可以看自己的显卡配置,用来选对应的英伟达开发包驱动。)

1. 下载 cuDNN:

NVIDIA CUDA® Deep Neural Network library (cuDNN) 大概54mb

https://developer.nvidia.com/rdp/cudnn-download

下载文件不稳定,多试几次(我下了三次)

这是个压缩文件包,解压缩后是三个文件夹,各有一个文件在里面。

如果6.0版本安装有错,请尝试5.1版本。感谢酒肉丸子的分享

2. 下载CUDA Toolkit 8.0: (我是下的8.0)

文件大小约1.2 G,下载文件不稳定,花了很久时间,重复用net 安装版

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

(C:\Users\xxxx\AppData\Local\Temp\CUDA 这个临时文件夹不用多虑,安装完了它会自动删除) 另外你要是装了360 就要不断“允许该进程所有操作 ”

理论上这个步骤会自动把路径安装到你的windows path里。

3. 把cuDNN的文件复制到CUDAToolkit 安装目录

应该在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0  (根据你的选择可能会有所不同)

总之你可以在那个目录下找到一堆文件夹,其中包括了那三个解压缩后的三个文件。

把那三个文件放到相同文件名的文件夹中。

万事俱备只欠重启!

重启电脑后,再次进入Anaconda Prompt

输入python进入

import tensorflow

看到几个 successfully opened

恭喜你GPU版安装完毕!

注意: 如果导入tensorflow时依然报错,请到path环境变量下确认路径

To use cuDNN with TensorFlow, the filecudnn64_5.dll must be in your%PATH%environment variable.

http://stackoverflow.com/questions/41007279/tensorflow-on-windows-couldnt-open-cuda-library-cudnn64-5-dll

一些其他的特殊情况:

https://github.com/tensorflow/tensorflow/issues/5968

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin


写完发现了两篇也不错的指导

http://www.itdadao.com/articles/c15a832274p0.html

https://zhuanlan.zhihu.com/p/24369784?refer=wjdml

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,835评论 4 364
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,598评论 1 295
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 109,569评论 0 244
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,159评论 0 213
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,533评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,710评论 1 222
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,923评论 2 313
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,674评论 0 203
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,421评论 1 246
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,622评论 2 245
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,115评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,428评论 2 254
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,114评论 3 238
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,097评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,875评论 0 197
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,753评论 2 276
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,649评论 2 271

推荐阅读更多精彩内容