安装tensorflow1.3 GPU Edition

1.事先需要准备的安装包

1.Ubuntu 16.04  下载链接:http://cn.ubuntu.com/download/

2.cuda 8.0,建议下载runfile格式 (现在官网下载的是cuda 9.0,但是目前的tensorflow 1.3并不能使用cuda 9.0, 会出现ImportError:libcusolver.so.8.0: cannot open shared object file: No such file or directory 的问题)  下载链接:https://developer.nvidia.com/cuda-toolkit-archive

3. cudnn 6.0 ,需要注册账号后才能下载(cudnn 5.1在tensorflow1.3上不再适用,所以下载cudnn 6.0)  下载链接:https://developer.nvidia.com/cudnn

NB:所有的下载链接的相应截图都是基于这篇文章的发布日期


2.安装

1. ubuntu16.04 不能用虚拟机安装,需要直接装在机器上(或者双系统)。 ubuntu系统盘制作工具下载链接:http://pan.baidu.com/s/1cpooQa     密码:lgdb   (系统盘做好后电脑设置u盘启动,然后就看着指示安装就可以了)

2.安装cuda 8.0 , 这里我用的是runfile的文件。在安装前要记得这个文件的下载目录,我一般都是放在 Downloads 下。 在终端输入以下命令:

sudo service lightdm stop      //关闭图形界面

CTRL+ ALT +F1       //进入字符串输入模式,按住这三个键调出命令行

输入用户名,密码,登陆。

cd 到 cuda 8.0 的文件目录

sudo sh cuda_8.0.44_linux.run         //这个语句不能直接复制,看你的cuda的命名是什么

等一会,然后会出现一个类似安装协议之类的东西,看完后选择 accept , 继续安装。之前看别人的安装方法的时候警告openGL不要安装,然后如果你之前没有装nvidia驱动的话(废话,新装的系统怎么会安装nvidia驱动),就需要安装nviidia-xconfig,否则可能会进入不了图形界面或者循环登陆。其他的差不多都选择y或者默认就可以了。

安装cuda时一定不要安装OpenGL;切记,否则有可能在安装完之后无法启动图形化桌面。

3. 安装 cudnn 6.0

其实不能称为安装 cudnn 6.0, 因为只是个简单的copy 的动作而已。

下载后解压,会有一个cuda的文件夹,然后就i是把其中的一些文件复制到cuda 8.0的对应目录下就可以了。

sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include

sudo cp cuda/lib64/libcudnn*/usr/local/cuda-8.0/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*

如果命令行复制失败,你也可以找到对应目录(命令行里有)直接copy 进去。

3.配置cuda环境变量

打开终端,输入以下命令即可。如果vim没有安装,则根据提示安装vim即可。命令行里的cuda-8.0可别直接复制了,如果你安装的是cuda-9.0 就改成cuda-9.0.

//打开配置文件

//按住 INSERT 进入编辑模式,输入以上两句语句在文件中,按ESC输入:wq,退出保存,

//刷新配置变量

sudo vim ~/.bashrc                  

export PATH=$PATH:/usr/local/cuda-8.0/bin

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64

export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-8.0/lib64

source ~/.bashrc     

到这里cuda 和cudnn已经安装完毕了,如果你没有遇到任何问题就按照流程走到了这一步,那么很恭喜。

4.安装tensorflow

我不得不吐槽一下,tensorflow官网的访问速度在国内访问实在是太慢了。确切来说,如果不翻墙的话根本访问不了。

但是不用担心,官网的安装指南固然详细,但是太详细了~我一开始看的时候是有点蒙的。tensorflow有很多不同的安装方式,这里我们选择native安装。

说到底,一张截图就可以搞定了。我把其中的一些命令copy下来,方便大家复制~

$sudo apt-get install python-pip python-dev# for Python 2.7

$sudo apt-get install python3-pip python3-dev# for Python 3.n

$pip install tensorflow# Python 2.7; CPU support (no GPU support) 

$pip3 install tensorflow# Python 3.n; CPU support (no GPU support) 

$pip install tensorflow-gpu# Python 2.7;  GPU support 

$pip3 install tensorflow-gpu# Python 3.n; GPU support

如果觉得上面的还是有点复杂,那么再简单一点吧。只有两条语句就够了(python3.n改成相应语句):

$sudo apt-get install python-pip python-dev  # for Python 2.7

$pip install tensorflow-gpu# Python 2.7;  GPU support

到这里就全部安装完毕,现在我们检测一下是否安装成功。

在终端输入以下命令:

python

import tensorflow as tf

hello=tf.constant('Hello, TensorFlow!')

sess=tf.Session()

print(sess.run(hello))

如果出现:Hello, TensorFlow!

那么很庆幸,安装成功了。当然大多数情况你没有这么顺利能够没有遇到任何麻烦就走到这一步的~

5.写篇文章用到的一些参考:

http://blog.csdn.net/appleml/article/details/76922476

http://www.linuxidc.com/Linux/2016-11/137561.htm

http://blog.csdn.net/wx7788250/article/details/77854217

http://www.jianshu.com/p/f459f4ab0d99

https://segmentfault.com/a/1190000008234390

网上虽有其他很多的文章写了如何安装tensorflow gpu 版本,但是 tensorflow 和cuda的更新让之前的文章很多方法并不适用 了,导致我在安装的时候遇到了一些麻烦。比如cuda 9.0 目前不适用于 tensorflow,但是很多文章给出的cuda下载链接都是直接下载cuda 9.0 的,导致我按照他们的方法安装后根本就失败了;然后tensorflow更新到1.3的版本后cudnn 5.1已经不适用了,可是现在搜到的文章还是说安装5.1的版本。还有一些文章强调gcc要降级到5.0以下的版本,但其实现在已经不用了~所以我才强调要注意文章给出链接和截图的日期。

为了让大家把精力放在研究上而非准备研究上,这是本文撰写的目的。

NB:文章转载需要告知作者

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