在Ubuntu 18.04安装tensorflow 1.12 GPU版本

在之前的深度学习中,我是在MAC上跑CPU版本的tensorflow程序,当数据量变大后,tensorflow跑的非常慢,在内存不足情况下,又容易造成系统崩溃,这个时候我觉得不应该浪费我的限制的微星游戏本,便想着拿来跑深度学习的代码。

1. 配置信息

我的老电脑配置如下:

  • CPU i5-4210M
  • 16G内存
  • GPU GTX 950M 显存2G
  • 128G SSD
  • Ubuntu 18.04

这个配置一般,但是为了不让我的mac发光发热,同时体验下GPU给深度学习的加速效果(虽然可能加不了多少速度)以及不想让老电脑荒废的心,我最终决定还是在上面安装Ubuntu18.04跑TensorFlow了。

2. 选择安装所需软件

TensorFlow官网中提到了需要安装以下软件才可以使用TensorFlow的GPU版本:
image

简单来说就是我们需要在Ubuntu18.04的环境下安装以下软件:

  1. NVIDIA的GPU驱动(nvidia drive-390)

  2. CUDA 9.0(不要下载CUDA 10.0,因为TensorFlow 1.12.0目前只支持到CUDA 9.0)

  3. cuDNN v7.3.1 for CUDA 9.0 (选当前最新的)

至于CUDA和cuDNN是什么自行谷歌,对应的下载页面和地址如下:

3. 安装NVIDIA驱动并降低gcc版本

cuda的官网 https://docs.nvidia.com/cuda/cuda-installation-guide-linux/#choose-installation-method 安装说明中已经列出了检查步骤。如果你的GPU在支持列表中,则我们只需要做如下操作:

3.1 安装GPU驱动程序。

Ubuntu在安装后,是给了一个叫Nouveau默认内建的驱动程序。我们需要安装GPU针对性的驱动程序,有篇文章教怎么在Ubuntu上安装合适的驱动:https://linuxconfig.org/how-to-install-the-nvidia-drivers-on-ubuntu-18-04-bionic-beaver-linux

简单来说,执行 ubuntu-drivers devices 命令得到推荐的nvidia驱动程序,

然后执行 sudo apt-get install nvidia-driver-390 (根据提示的驱动程序而定)即可安装完成。

注意安装完成后需要重启。

重启完成后执行nvidia-smi,如果出现类似于下图所示的信息则说明驱动安装成功:

image

3.2 安装gcc-4.8和g++4.8

sudo apt-get install gcc-4.8
ll /usr/bin |grep gcc #若存在高版本的gcc,则继续
cd /usr/bin
sudo mv gcc gcc.bak #备份
sudo ln -s gcc-4.8 gcc  #重新指向

执行gcc --version 可检查是否为4.8版本,g++可执行同样的操作。

4. 安装CUDA和cuDNN

进入CUDA9.0的下载页,选择安装类型为runfile(local),可发现有一个基础安装包,和三个补丁包,都下载下来,然后依次执行:

sudo sh cuda_9.0.176_384.81_linux.run

安装时会以提问的方式,一路默认即可,当第二个询问是否安装新驱动程序时,选择否,其余均是yes。然后再以同样的方式安装以下补丁包。

sudo sh cuda_9.0.176.1_linux.run
sudo sh cuda_9.0.176.2_linux.run
sudo sh cuda_9.0.176.3_linux.run 

结束后,可以在/usr/local/cuda-9.0 发现有安装文件,执行sudo gedit /etc/profile或者sudo gedit ~/.bashrc/etc/profile~/.bashrc的文件后面添加环境变量:

export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}  

然后执行source ~/.bashrc或者source /etc/profile让刚刚修改的环境变量生效。然后执行nvcc -V来验证CUDA是否完成安装。

安装好CUDA9.0后,接下来就安装cuDNN,进入cuDNN下载页选择 cuDNN v7.3.1 Library for Linux 下载。

​ 执行如下命令:

tar -xzvf cudnn-9.0-linux-x64-v7.tgz

进入解压目录,执行:

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

这样就安装完成了。

5. 安装pyenv

这里我采用了pyenv的方式来隔离不同环境下的python。

首先安装curl,它一个命令行式的下载工具

sudo apt install curl

然后下载pyenv的安装包,pyenv的GitHub链接:pyenv

curl -L https://raw.githubusercontent.com/yyuu/pyenv-installer/master/bin/pyenv-installer | bash

编辑~/.bashrc:

sudo gedit ~/.bashrc

在该文件中添加以下内容(直接复制即可):

export PYENV_ROOT="$HOME/.pyenv"
export PATH="$PYENV_ROOT/bin:$PATH"
eval "$(pyenv init -)"
eval "$(pyenv virtualenv-init -)"</pre>

终端执行以下命令安装python的依赖包

sudo apt-get update
sudo apt-get install make build-essential libssl-dev zlib1g-dev
sudo apt-get install libbz2-dev libreadline-dev libsqlite3-dev wget curl
sudo apt-get install llvm libncurses5-dev libncursesw5-dev

安装python 3.6.5

pyenv install 3.6.5

安装过程可能很缓慢,一个简便方法,只需要在python的官网(此处给出python3.6.5的下载地址)下载你需要的python版本的tar.xz文件然后放到 ~/.pyenv/cache中然后再执行pyenv install [version]就可以了,cache文件夹可能不存在,请自行新建。

image

pyenv的常用命令

image

上图是官方文档中的例子,如果想要查看完整命令列表,可以点击查看pyenv命令列表

在终端执行pyenv global 3.6.5即可将3.6.5设置为系统默认的python环境

6. 安装tensorflow-gpu

终端中输入以下命令即可安装:

pip install --index-url http://pypi.douban.com/simple --trusted-host pypi.douban.com --upgrade tensorflow-gpu

7. 验证结果

利用终端新建一个.py文件

touch test.py

然后利用Visual Studio Code打开该文件

code test.py

输入

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

然后执行该脚本python test.py即可得到包含有GPU输出信息的正确结果

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

推荐阅读更多精彩内容