Ubuntu操作系统安装Nvidia GPU驱动

本章以Ubuntu 20.04系统为例，介绍Nvidia GPU驱动的安装方法。

1.1 禁用nouveau驱动

禁用开源的nouveau驱动，避免和Nvidia GPU驱动冲突。

cat >> /etc/modprobe.d/blacklist-nouveau.conf << EOF
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
EOF

下面第2小节使用deb包安装驱动 和第3小节使用run文件安装驱动 二选一即可。

注意安装Nvidia GPU驱动需要在非桌面环境运行。如果处于桌面环境中，可以执行命令systemctl set-default multi-user.target，然后重启来切换到虚拟终端界面。稍后可以执行命令systemctl set-default graphical.target，然后重启再切换回桌面环境中。

1.2 使用deb包安装驱动（和1.3二选一）

1.2.1 下载驱动

访问页面

https://www.nvidia.cn/content/DriverDownloads/confirmation.php?url=/tesla/470.129.06/nvidia-driver-local-repo-ubuntu2004-470.129.06_1.0-1_amd64.deb&lang=cn&type=Tesla

点“同意并开始下载”以下载驱动。

或者直接使用下述链接下载驱动。

https://cn.download.nvidia.com/tesla/470.129.06/nvidia-driver-local-repo-ubuntu2004-470.129.06_1.0-1_amd64.deb

1.2.2 安装驱动

假设下载好的驱动已经放到了/root/nvidia-driver-local-repo-ubuntu2004-470.129.06_1.0-1_amd64.deb。按下面指令安装Nvidia GPU驱动。

cat > /etc/apt/preferences.d/nvidia <<EOF
Package: *
Pin: release o=NVIDIA
Pin-Priority: 550
EOF
dpkg -i /root/nvidia-driver-local-repo-ubuntu2004-470.129.06_1.0-1_amd64.deb
apt-key --keyring /usr/share/keyrings/nvidia-driver-local-D9CB5EF8-keyring.gpg add /var/nvidia-driver-local-repo-ubuntu2004-470.129.06/D9CB5EF8.pub
apt-get update
apt-get install -y cuda-drivers
apt-get purge -y nvidia-driver-local-repo-ubuntu2004-470.129.06
apt-key --keyring /usr/share/keyrings/nvidia-driver-local-D9CB5EF8-keyring.gpg del D9CB5EF8
rm -vf /etc/apt/preferences.d/nvidia

1.3 使用run文件安装驱动（和1.2二选一）

1.3.1 下载驱动

使用下述链接下载驱动：

https://download.nvidia.com/XFree86/Linux-x86_64/470.129.06/NVIDIA-Linux-x86_64-470.129.06.run

1.3.2 安装驱动

假设下载好的驱动放在/root/NVIDIA-Linux-x86_64-470.129.06.run

apt -y install build-essential #添加必要依赖
chmod +x /root/NVIDIA-Linux-x86_64-470.129.06.run
/root/NVIDIA-Linux-x86_64-470.129.06.run

1.3.3 验证驱动是否正常安装

nvidia-smi

出现类似如下界面，说明驱动安装完成。