ubuntu18.04安装CUDA、cuDNN-华为MagicBook Pro

Tensorflow的版本需要和CUDA、cuDNN版本相对应,具体的对应关系可以查看https://www.tensorflow.org/install/source中给出的对应列表。我要跑的一个开源程序需要用到的Tensorflow版本为1.10.0,查阅可知,需要的CUDA版本为9、cuDNN版本为7。

安装NVIDIA驱动

CUDA的安装分在线和离线两种方式。离线方式下载的安装包自带NVIDIA 驱动,在线方式需要自己安装NVIDIA 驱动。由于我以后可能会同时安装并存多种版本的CUDA,所以我选择手动安装NVIDA驱动然后在线安装CUDA的方法。如果你选择的是离线安装CUDA的话只需要下载对应的离线CUDA包进行安装就可以了,不需要自己手动安装NVIDIA驱动。
NVIDIA驱动的安装过程参考了博客https://www.vicw.com/groups/water/topics/200的指导。

安装CUDA10.0

下载

从网站https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=debnetwork下载与Ubuntu18.04系统对应的在线安装包,下载后的文件名为cuda-repo-ubuntu1804_10.0.130-1_amd64.deb

安装

按照下载页面给出的安装命令进行安装

sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo apt-get update`
sudo apt-get install cuda-10-0`

上述命令运行过程中可能出现依赖的包不会被安装的问题,这时循环使用apt updateapt upgrade命令,直到所有的包都达到最新就可以了。如果这个过程中出现了明明有包需要更新,但是apt upgrade命令就是没办法更新它们的情况,那么用apt dist-upgrade命令来代替apt upgrade命令,并结合使用apt autoremove来进行更新。
哎呀网络不行啊,要下载1天零5个小时。
使用sudo apt install cuda-libraries-10-0来安装所有的library包。

修改环境变量

编辑/etc/profile文件,在最后加上下面这一行:

export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}

验证安装结果

重启系统,输入如下命令

cat /proc/driver/nvidia/version

返回如下结果


Screenshot from 2020-03-19 14-12-10.png

输入如下命令

nvcc -V

返回如下结果


Screenshot from 2020-03-19 14-13-20.png

证明安装成功。

安装cuDNN

https://www.tensorflow.org/install/source#linux表明版本为7.4的cuDNN经过测试,是可以和CUDA10.0、Tensorflow1.14.0一起正常使用的,所以我选择下载安装cuDNN7.4,具体来说,是cuDNN 7.4.2,从网站https://developer.nvidia.com/rdp/cudnn-archive上下载(你需要注册NVIDIA的账号并登录)。

下载

下载如下图中的三个包


Screenshot from 2020-03-19 14-22-12-2.jpg

安装runtime library

sudo dpkg -i libcudnn7_7.4.2.24-1+cuda10.0_amd64.deb 

安装developer library

sudo dpkg -i libcudnn7-dev_7.4.2.24-1+cuda10.0_amd64.deb 

安装code samples和cuDNN Library User Guide

sudo dpkg -i libcudnn7-doc_7.4.2.24-1+cuda10.0_amd64.deb

验证安装结果

cp -r /usr/src/cudnn_samples_v7/ ./
cd cudnn_samples_v7/mnistCUDNN
make clean && make
 ./mnistCUDNN

出现test passed信息,表明安装成功。

安装libcupti-dev library

sudo apt install libcupti-dev

安装GPU版本的Tensorflow-1.14.0

pip3 install --user tensorflow-gpu==1.14.0

验证安装结果

运行如下代码

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

出现如下信息

2020-03-19 15:41:41.961101: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1326] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 986 MB memory) -> physical GPU (device: 0, name: GeForce MX250, pci bus id: 0000:02:00.0, compute capability: 6.1)
Device mapping:
/job:localhost/replica:0/task:0/device:XLA_GPU:0 -> device: XLA_GPU device
/job:localhost/replica:0/task:0/device:XLA_CPU:0 -> device: XLA_CPU device
/job:localhost/replica:0/task:0/device:GPU:0 -> device: 0, name: GeForce MX250, pci bus id: 0000:02:00.0, compute capability: 6.1
2020-03-19 15:41:41.961625: I tensorflow/core/common_runtime/direct_session.cc:296] Device mapping:
/job:localhost/replica:0/task:0/device:XLA_GPU:0 -> device: XLA_GPU device
/job:localhost/replica:0/task:0/device:XLA_CPU:0 -> device: XLA_CPU device
/job:localhost/replica:0/task:0/device:GPU:0 -> device: 0, name: GeForce MX250, pci bus id: 0000:02:00.0, compute capability: 6.1

表明GPU版本的TensorFlow1.14.0安装成功。

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

推荐阅读更多精彩内容