众所周知,Ubuntu系统自带的显卡驱动是开源的显卡驱动。当我们在配置Ubuntu深度学习工作站时,通常需要将这个开源的显卡驱动更换为NVIDIA的官方驱动。
而根据工作站的显卡配置以及项目所需的CUDA版本,我们也需要切换到对应版本的显卡驱动。然而在Ubuntu默认的软件与更新
里一般只能找到很少的几个NVIDIA专有驱动,这就导致我们可能没法顺利配置我们想要的深度学习开发环境。
为了解决这个问题,这里就讲一下如何在软件与更新
当中配置第三方仓库PPA,并且通过第三方仓库安装NVIDIA显卡驱动。
一、使用国内软件源加速
在原生Ubuntu系统中执行sudo apt-get update
或者sudo apt-get upgrade
等命令时,往往命令执行速度是比较慢的。原因在于许多依赖包都是直接通过国外的源下载的,这里可以手动设置国内的镜像,步骤如下:
1、打开sources.list这个文件
sudo gedit /etc/apt/sources.list
2、清空里面的内容,把以下的源添加进去
# 阿里云源
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-backports main restricted universe multiverse
##測試版源
deb http://mirrors.aliyun.com/ubuntu/ bionic-proposed 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-backports main restricted universe multiverse
##測試版源
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse
# 清华大学源
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
##測試版源
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse
# 源碼
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
##測試版源
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse
3、更新软件列表
添加好国内软件源之后,执行命令更新软件列表:
sudo apt-get update
添加第三方仓库PPA
步骤如下:
1、执行以下命令添加显卡驱动的第三方仓库:
sudo add-apt-repository ppa:graphics-drivers/ppa
2、更新软件列表
sudo apt update
二、禁用开源驱动nouveau
前面提到,Ubuntu默认是使用开源驱动nouveau的。因此在安装NVIDIA驱动前需要禁用nouveau,方法如下:
1、连续执行以下两条命令
sudo bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
sudo bash -c "echo options nouveau modeset=0 >> /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
这两条命令将在文件/etc/modprobe.d/blacklist-nvidia-nouveau.conf
中写入禁用nouveau的配置,当然也可以直接找到这个文件用编辑器(vim或者gedit)把配置写进去。
2、然后可以执行以下命令验证以上的指令输入到/etc/modprobe.d/blacklist-nvidia-nouveau.conf
这个文件中:
cat /etc/modprobe.d/blacklist-nvidia-nouveau.conf
如果显示以下信息就表示已经成功禁用了:
blacklist nouveau
options nouveau modeset=0
三、正式安装NIVDIA驱动
1、卸载旧的NVIDIA驱动(如果需要的话):
sudo apt-get purge nvidia*
2、把显卡驱动加进PPA
sudo add-apt-repository ppa:graphics-drivers
sudo apt-get update
3、查找最新的NVIDIA驱动
sudo apt-cache search nvidia
4、通过以下指令查看推荐的驱动版本
ubuntu-drivers devices
5、打开系统里的Ubuntu软件
,然后点击左上角的“软件和更新”,点击“附加驱动”,选择其中一个专有驱动,然后点击进行“应用更改”。安装完之后系统会要求用户进行重启,这里按指示执行重启操作即可。
6、重启成功后可以查询设置
里的详细信息
,驱动安装正确的话应该能看到NVIDIA显卡的相关信息。