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 下載驅動

訪問頁面

//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

點“同意并開始下載”以下載驅動。

或者直接使用下述鏈接下載驅動。

//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 下載驅動

使用下述鏈接下載驅動：

//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

出現類似如下界面，說明驅動安裝完成。