服务器Ubuntu18.04安装Tensorflow-GPU 2.2.0

1. 更换Ubuntu18.04的apt源

1.1 备份apt源文件

sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak

1.2 修改apt源文件,替换成国内源

sudo vim /etc/apt/sources.list

注:清空 sources.list 文件后,粘贴国内源(下面是国内源,选择一个粘贴)

阿里云源:
deb http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse
清华源:
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse
中科大源:
deb https://mirrors.ustc.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb https://mirrors.ustc.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb https://mirrors.ustc.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
deb https://mirrors.ustc.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb https://mirrors.ustc.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse
deb-src https://mirrors.ustc.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb-src https://mirrors.ustc.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb-src https://mirrors.ustc.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
deb-src https://mirrors.ustc.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb-src https://mirrors.ustc.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse

1.3 同步 sources.list 和 sources.list.d 的源索引

sudo apt update

1.4 升级已安装的所有软件包

sudo apt upgrade

注:升级之后的版本就是本地索引里的,因此,在执行 upgrade 之前一定要执行 update, 这样才能是最新的。

2. 安装显卡驱动

2.1 禁用系统默认显卡驱动

2.1.1 打开黑名单

sudo vim /etc/modprobe.d/blacklist.conf

2.1.2 将下列代码填入文件末尾:

# for nvidia display device install
blacklist vga16fb

blacklist nouveau

blacklist rivafb

blacklist rivatv

blacklist nvidiafb

2.1.3 保存,然后更新initramfs:

sudo update-initramfs -u

2.1.4 重启电脑

sudo reboot

5.查看是否禁用成功,无输出则禁用成功

 lsmod | grep nouveau

注:这里直接从 ubuntu库 中安装显卡驱动

2.2 查看显卡型号

ubuntu-drivers devices

驱动后面 recommended 的是系统推荐安装的版本

image.png

注:如果没有这个包可以按照提示进行安装:
sudo apt-get install ubuntu-drivers-common

2.3 安装系统推荐的显卡驱动

sudo ubuntu-drivers autoinstall

注:也可以使用 apt 命令安装自己想要安装的版本,比如我想安装 450 这个版本号的版本,终端输入:sudo apt install nvidia-driver-340 就自动安装了

2.4 安装完成后重启

sudo reboot

2.5 重启完成查看显卡信息,显示如下图内容就安装成功了

nvidia-smi
image.png

3. 安装 Minicoda 或 Anaconda

为什么使用 conda 安装?
使用 conda 安装 tensorflow 会自动安装依赖,包括自动安装对应版本的 cudacudnn,避免版本不对造成不必要的麻烦

使用 Anaconda 还是 Miniconda?
Anaconda:是一个开源的Python发行版本,包含了conda、python等180多个科学包及其依赖项。因为包含了大量的科学包,所以Anaconda的安装包比较大
Miniconda:这个较小的发行版,大部分包都要自己后期安装
根据自己需求下载 Anaconda 或者 Miniconda

下载地址
国外下载 Anaconda 比较慢,下面是清华源 Anaconda 镜像下载地址
清华源 Anaconda 镜像下载地址
清华源 Miniconda 镜像下载地址
找到最新版根据自己系统进行下载:
Anaconda:

image.png

Miniconda:
image.png

3.1 安装

sh Miniconda3-latest-Linux-x86_64.sh

3.2 查看协议,同意协议进行安装

3.2.1 点击 回车 查看协议,空格 翻页

image.png

3.2.2 输入 yes 同意协议继续安装

image.png

3.2.3 默认安装路径是 ~/miniconda3 (anaconda的是 ~/anaconda),也可以自己输入路径进行安装,这里就默认路径不做改动,点击回车继续安装

image.png

3.2.4 输入 yes 初始化 conda


image.png

3.2.5 安装完成


image.png

3.2.6 重启
sudo reboot

4. 创建虚拟环境,并安装 tensorflow-gpu 2.2.0

tensorflow-gpu 版本对应

image.png

4.1 conda 添加清华源,并设置默认用新的源下载

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

注:如果安装包没有使用添加的镜像而报网络错误,可以使用下面命令删除默认源:
conda config --remove channels defaults

4.2 创建虚拟环境

sudo conda create -n 虚拟环境名称 python=3.7
下面 myenv 是自定义的虚拟环境名称,python版本3.7(tensorflow-gpu 2.2.0 支持 python3.5-3.7)

conda create -n myenv python=3.7

4.3 进入虚拟环境

conda activate myenv

前面显示 虚拟环境名称 就说明已经进入虚拟环境

image.png

4.4 查看支持的 tensorflow-gpu 版本

 conda search tensorflow-gpu
image.png

4.5 安装 tensorflow-gpu 2.2.0

conda install tensorflow-gpu=2.2.0

注:如果由于网络原因,比较大的包没有装上,可以重试几次 conda install tensorflow-gpu=2.2.0 命令,会自动安装没有按上的包

4.6 安装完成

image.png

4.7 测试是否安装成功

检测tf是否安装成功:

$ python3
>>> import tensorflow as tf
>>> hello = tf.constant('Hello World!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
Hello World!
>>> a = tf.constant(8)
>>> b = tf.constant(2)
>>> print(sess.run(a + b))
10
>>>

使用 tf.test.is_gpu_available() 来测试TF是否可以使用GPU来加速计算,如果输出True则代表成功。

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

推荐阅读更多精彩内容