win10安装tensorflow-cpu和tensorflow-gpu

SZU

介绍

tensorflow使用之前的安装和环境配置是头等大事,只有这个解决了,才能进行后面的学习和使用。本文介绍了在win10 操作系统上面安装tensorflow-CPU和tensorflow-GPU的环境。



说在安装之前

  • tensorflow需要先安装anaconda,然后再在anaconda中安装tensorflow,conda install tensorfow,这里最好先下载镜像,然后安装,国内容易被墙,很多时候安装一大半就失败了。
  • 安装anaconda,再安装其cpu或者GPU版本。最后再在spyder中使用。
  • 然而,对于习惯使用pycharm而非anaconda的朋友来说,安装tensorflow是一件多么不幸的事情呀,pip install tensorflow,至少我没有成功过,可以安装完成anaconda后,在pycharm中调用anaconda中的解释器。
  • 对于anaconda中是否需要再安装一个虚拟环境,在这个环境里再安装tensorflow这个问题,我觉得完全没有必要,虽然网上很多贴子都是这么做的,anaconda已经安装这么多包了,不在乎多这么一个了。


一、win10 搭建tensorflow-CPU环境

已经在另一篇博客中介绍了,pycharm+anaconda+tensorflow(CPU)

二、win10 搭建tensorflow-GPU环境

cpu只能解决一些运算量很小的问题,学习还可以,但真需要使用时,就捉襟见肘了。这里就要用到GPU来加速了。在搭建tensorflow-gpu环境之前有几个问题不得不注意:

  • 1、cuda9.0+cudnn9.0+tensorflow-gpu1.9.0,三者版本需要对的上
  • 2、cunda9.0的安装路径和环境变量的配置问题
  • 3、tensorflow1.9.0和numpy的不兼容问题,numpy==1.14.0降低到numpy==1.13.0
  • 4、使用清华镜像源会出现的问题

1、环境

  • 操作系统:win10
  • GPU:GTX1050+CUDA9.0+cuDNN9.0
  • IDE:Pycharm
  • 框架:tensorflow-gpu1.9.0
  • 解释器:Python3.6
    (强烈建议anaconda3,又方便又强大,下载太慢可以找镜像)

2、安装顺序

(1)首先安装CUDA9.0 官方下载地址

现在最新的已经是9.2版本了。而我选择9.0版本,是因为网上说,tensorflow的windows-gpu还不支持CUDA的新版本。再加上网上安装9.0版本的教程都成功了,我就偷偷懒。
按照要求,一路绿灯。
note:这里的安装路径选用系统默认,最后肯定会安装到C盘

(2)CUDA9.0 环境变量配置

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin

(3)安装cuDNN9.0 官网下载地址

下载搭配CUDA9.0的cudnn ,一定要注意搭配9.0。需要注册英伟达的账号,这个过程可能很痛苦,国外网站使用不易。

下载好之后,解压,分别将.h .lib 和.dll 文件分别拷贝到cuda的include, lib/x64, bin 文件夹下。其实也就是对应的文件夹。

到这里就安装好了CUDA9.0及其对应的cudnn。

(3)安装tensorflow-GPU

为了不相干扰,先删除之前安装的是cpu版本。

>> pip uninstall tensorflow

然后安装清华镜像源中的程序

>> pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.9.0

添加到path当中

(4) 验证

在命令行中,进入python,并输入以下代码:

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

输出:

b’ Hello, TensorFlow!

安装完成

3、遇到的大坑

(1)镜像源带来的版本问题

https://mirrors.tuna.tsinghua.edu.cn/
https://mirrors.tuna.tsinghua.edu.cn/help/pypi/
注意两者的区别,前者不是最新的安装包,很容易出现问题

以前使用清华镜像源来安装软件,能大大提升下载速度,但是遇到一个很严重的问题,清华镜像源上的版本问题,这里使用的是cuda9.0+cuDNN9.0,自然tensorflow-gpu也应该是1.9.0版本,而清华镜像源上的是1.1.0版本,根本就不行。这里要先解除镜像源链接,再重新安装。

  • 添加 Anaconda 仓库的镜像源:
>> conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
>> conda config --set show_channel_urls yes

添加之后,就会从镜像源中直接下载,速度是非常快的

  • 查看镜像源:
conda config --show
  • 删除镜像源:
conda config --remove channels 'https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/' 

对于一些镜像源出现的版本限制,可以删除此镜像源

(2)tensorflow1.9.0及其他高版本对numpy的不兼容问题

运行的时候会出现如下错误类型:

.local/lib/python2.7/site-packages/h5py/init.py:36: FutureWarning: Conversion of the second argument of issubdtype from float to np.floating is deprecated. In future, it will be treated as np.float64 == np.dtype(float).type.
from ._conv import register_converters as _register_converters

解决方法:这是因为numpy高版本与h5py不兼容导致,可以用 pip install numpy==1.13.0 安装旧版本的numpy解决

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

推荐阅读更多精彩内容