deepin系统下搭建GPU深度学习环境

最近折腾了一个deepin系统,尝试配置了一下pytorch和tf对应的gpu版本,走了不少弯路。其实整个流程不出错误的话非常简单,这里简单做一下备忘。

显卡驱动配置

目前新版本的deepin系统都自带一个显卡驱动管理器,被这个东西坑哭了。使用这个安装的闭源驱动和英伟达官方的驱动是有区别的,虽然都可以nvidia-smi命令查看驱动状态,但是如果安装了这个驱动,后面的步骤无法识别的。

1 下载官方驱动

去英伟达官网,根据型号,下载对应的驱动。长寿命和短寿命不太清楚,个人下了短寿命版本。

2 禁用nouveau驱动

这一步是看大部分人都有提到的,但是我个人没碰到这个情况。
先查看是否已经禁用

lsmod | grep nouveau

如果屏幕没输出,说明已经禁用。不然的话执行以下操作:

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

文本最后添加

blacklist nouveau
options nouveau modeset=0

执行

sudo update-initramfs -u

重启之后再次查看,如果没有输出,说明成功。 如果还是有问题,建议移步其他地方查看解决方案。

3 清理现存驱动

如果没有装驱动,可以直接跳过这个步骤。

3.1 deb安装驱动

如果用deb安装了驱动,系统的那个显卡驱动管理就是属于此范畴。

sudo apt-get remove --purge nvidia*

zsh可能会找不到对应命令,要切回到bash

3.2 已安装老版本英伟达驱动

sudo apt-get remove nvidia-driver nvidia-kernel-dkms glx-alternative-nvidia

sudo nvidia-uninstall
sudo nvidia-installer --uninstall

4 安装驱动

首先进入控制台界面(ALT+CTRL+F4)。
关闭图形界面

sudo service lightdm stop

进入下载的驱动文件路径

# 添加权限
sudo chmod a+x NVIDIA-Linux-x86_64-430.26.run
# 安装
sudo sh ./NVIDIA-Linux-x86_64-430.26.run

安装成功后打开图形界面

sudo service lightdm start

可能要重启才能看到效果。
查看驱动信息

nvidia-smi

理论上最新的驱动,显示的版本型号应该都是400+,cuda10.0+。

使用conda安装环境

对于个人开发者而已,这里已经到了最后一步了。
目前网上大部分的流程是安装cuda和cudnn这两个。 但是如果是个人开发使用,完全不必。
用conda创建对应环境。这一步不会的网上教程一大堆。

# 安装pytorch
conda install pytorch-gpu
# 安装tf
conda install tensorflow-gpu

结束。

这两种方式会直接安装最新的,但如果cuda版本不支持的话,需要去官网查看一下,安装对应的cuda版本。

如果不用conda的话,个人认为直接 pip install xxx-gpu理论上也可以。
按部就班安装的话后面也太多坑。
deepin的gcc是6.0,安装cuda9.0的话不需要降级。
剩下的基本就是下载对应版本的内容,安装就可以。

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

推荐阅读更多精彩内容