本文总共1738个字,阅读需6分钟,全文加载时间:2.900s,本站综合其他专栏收录该内容! 字体大小:

文章导读:由于工作相关原因,需要用到Ubuntu系统进行AI算法测试以及训练深度学习模型等工作。公司有服务器版的Ubuntu系统,但终究是公司资产,作为小白自然是不敢轻易折腾。 因此,在自己的笔记本电脑上安装了Ubuntu18.04,……各位看官请向下阅读:

由于工作相关原因,需要用到Ubuntu系统进行AI算法测试以及训练深度学习模型等工作。公司有服务器版的Ubuntu系统,但终究是公司资产,作为小白自然是不敢轻易折腾。

因此,在自己的笔记本电脑上安装了Ubuntu18.04,与Windows10组成了双系统。关于如何安装win10+Ubuntu双系统,这里不展开,相关教程参考链接:https://www.cnblogs.com/masbay/articles/10745170.html

注意:如果是在Windows系统上使用VMware、Hyper等虚拟机工具(包括win10自带支持的Ubuntu子系统)安装的Ubuntu系统,无法挂载安装GPU驱动。

前期准备:

1.禁用BIOS中的secure boot

根据其他使用者的描述,如果不禁用,将会出现某些奇怪的问题。因为我在安装Ubuntu时就已经禁用了windows系统的secure boot,所以在后续安装显卡驱动时并没有遇到问题。

2.将源换为国内源

Ubuntu本地源的速度众所周知,极大影响后续进行的更新下载工作,因此需要换成国内源,比如:清华源、阿里源、豆瓣源等。由于我使用图形界面版Ubuntu,即打开“软件和更新”,在第一个顶部菜单下的“下载自”下拉选择任何一个国内源。如下图,我使用的是清华源。

如果使用server版Ubuntu,即没有图形界面需要使用ssh登录,则执行命令换源。由于后续我需要安装并且主要使用annconda,因此换源命令可直接使用conda命令。注意:ubuntu换源可以在安装更新Ubuntu软件或包时加快下载速度(例如安装显卡驱动),annconda换源则是在conda环境下进行相关操作。

ubuntu换源命令参考:https://blog.csdn.net/maizousidemao/article/details/79127695

conda换源命令如下:

conda config --add channels https://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

3.更新Ubuntu软件包

sudo apt-get updatesudo apt-get upgrade

4.禁用nouvean

小白表示不清楚这是个什么,但大神们都说需要禁用,那就跟着大神的步伐走。

输入命令查看是否被禁用,如果enter返回后无结果显示,则表明已被禁用。

lsmod | grep nouveau

否则,按以下步骤执行禁用:

#打开文件sudo gedit /etc/modprobe.d/blacklist.conf#在文件末尾添加blacklist nouveauoptions nouvean modeset=0#保存文件,更新重启sudo update-initramfs -usudo reboot

开始安装:

1.命令查看本机适合的显卡驱动版本

这个步骤只适用于目前单纯挂载显卡驱动,后续安装cuda、深度学习框架等,需根据cuda版本或框架环境要求针对性地选择驱动版本。如下图,带有“recommended”的驱动版本就是适合本机的。

ubuntu-drivers devices

2.安装NVIDIA显卡驱动

方法(1):自动安装,一行命令搞定,系统会自主选择安装当前适合本机的驱动版本(本机测试成功)

sudo ubuntu-drivers autoinstall

方法(2):选择安装,自己选择想要安装的版本号(本机没有测试)

sudo apt-get install nvidia-driver-460

方法(3):PPA安装(本机测试失败,可能是网络原因,无法添加PPA)

sudo add-apt-repository ppa:graphics-drivers/ppasudo apt-get updatesudo apt-get install nvidia-driver-460

3.重启验证

sudo rebootnvidia-smi

以上内容由优质教程资源合作伙伴 “鲸鱼办公” 整理编辑,如果对您有帮助欢迎转发分享!

你可能对这些文章感兴趣:

发表评论

您的电子邮箱地址不会被公开。 必填项已用*标注