Ubuntu上的深度学习环境配置

本文适用于配置 《动手学深度学习》书中的 Linux + NVIDIA 显卡环境。因文中对于GPU的安装描述略显简单,故结合个人经验拾遗补缺。

参考文献:

https://blog.csdn.net/xunan003/article/details/81665835

https://blog.csdn.net/u010801439/article/details/80483036

https://www.jianshu.com/p/52a49bcb2fac

1. 准备一台带有GPU的电脑(废话)

2. 安装显卡驱动和CUDA, cuDNN。

Ubuntu18.04 默认安装,一路回车即可。后续步骤参照我的显卡(GTX 1060 3GB显存)安装驱动程序等。

2.1 驱动安装

首先,禁用第三方开源的驱动程序nouveau,避免安装时登录界面循环问题。

禁用之前先查看nouveau驱动是否存在:

lsmod | grep nouveau

若显示为空可直接跳过后续步骤直接下载新版驱动。否则,执行命令编辑 blacklist-nouveau.conf 文件。

sudo vi /etc/modprobe.d/blacklist-nouveau.conf

修改后的文件内容如下:

更新系统,然后重启(务必)

sudo update-initramfs -u

重启后,验证nouveau是否已禁用

lsmod | grep nouveau

然后,官网下载相应驱动并安装:

下载完毕后,将文件拷贝到任意目录下并赋予运行权限。

sudo chmod  a+x NVIDIA-Linux-x86_64-xxx.run

图形界面下按“ctrl+alt+f1”进入命令行界面,在命令行界面下输入以下命令,关闭图形界面:

sudo service lightdm stop      

如安装过其他版本驱动,则卸载掉原有驱动:

sudo apt-get remove nvidia-*  

执行安装,附带以下参数:

sudo ./NVIDIA-Linux-x86_64-xxx.run -no-x-check -no-nouveau-check -no-opengl-files

-no-x-check:安装驱动时关闭X服务

-no-nouveau-check:安装驱动时禁用nouveau

-no-opengl-files:只安装驱动文件,不安装OpenGL文件

安装过程中的选项,建议选择如下(选错了也没关系,多试几次就行了):

1. DKMS:No

2. 32-bit 兼容:No

3. NVIDIA-xconfigutility :Yes

安装结束后,挂载驱动:

modprobe nvidia

执行命令检查驱动是否安装成功:

nvidia-smi

如果出现如下提示,则说明安装成功:

最后建议重启系统

2.2 安装CUDA和cuDNN

进入官网,选择CUDA版本并进行下载:

拷贝文件到任意路径,执行命令安装:

sudo sh ./cuda_10.xxx_linux.run

需要注意的是,因之前已经安装过显卡驱动程序,故在提问是否安装显卡驱动时选择No,其他默认即可。

安装完毕后,根据提示添加两条路径到 .bashrc 文件中(注意CUDA版本):

sudo vi ~/.bashrc

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

export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

至此,CUDA安装完毕,可执行以下命令测试,若最后提示“pass”则安装成功,否则需要卸载后重新安装。

nvcc --version

cd /home/hyperly/NVIDIA_CUDA-10.0_Samples/1_Utilities/deviceQuery

make

./deviceQuery

之后安装cuDNN,需要注册后才可下载。

点击官网,注册账号后下载:

下载完毕后,进入文件夹,将 cudnn-10.1-linux-x64-v7.5.1.10.tgz 压缩包先解压,然后将其中的内容复制到 /usr/local 的CUDA安装文件夹里面。

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*

至此,cuDNN的安装完成。

此外,若需要删除CUDA和cuDNN,可使用如下命令:

sudo /usr/local/cuda/bin/uninstall_cuda_10.0.pl

sudo rm -rf /usr/local/cuda

3. 配置GPU版本的运行环境

3.1 参照书中完成前两步的配置,即安装Miniconda和下载解压最新代码。

3.2 使用conda直接创建GPU版本的虚拟运行环境。

此处需根据显卡驱动版本,改变为依赖为GPU版本的MXNet。使用文本编辑器打开本书的代码所在根目录下的文件environment.yml,将里面的字符串“mxnet”替换成对应的GPU版本。例如,本机上装的是10.0版本的CUDA,则将该文件中的字符串“mxnet”改为“mxnet-cu100”。同时还需要更新MXNet的对应版本(若版本不对则会报错,直接从出错信息中选择一个最新的版本号然后修改environment.yml文件即可)。

保存文件后退出,例如本机最后的配置信息如下:

3.3 激活环境并使用Jypyter。


P.S. 如果重新下载了代码,则需要按照书中所述,卸载MXNet, 更新environment.yml文件并重新激活环境。

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