0. 引言
在Ubuntu上安装显卡驱动有两种方式,一种是通过直接下载对应的显卡驱动文件进行安装,另一种是通过安装CUDA Toolkit,随着CUDA Toolkit附带安装。本文将介绍这两种安装方式,二选一即可。
1. 禁用nouveau驱动
禁用开源的nouveau驱动,避免和Nvidia GPU驱动冲突。
cat >> /etc/modprobe.d/blacklist-nouveau.conf << EOF
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
EOF
2. 直接下载
到下载 NVIDIA 官方驱动 | NVIDIA网站下,选择对应的显卡型号,CPU架构,系统版本,找到对应的显卡驱动文件进行下载。
下载完成会得到一个.deb文件,将该.deb文件上传到服务器上。
假设下载好的驱动文件放在/root目录下。按下面的指令对.deb文件进行解压。
dpkg -i /root/<刚刚下载的.deb文件>
解压完成之后,屏幕上会出现一个cp指令,运行这行命令
sudo cp /var/nvidia-driver-local-repo-ubuntu2004-570.133.20/nvidia-driver-local-C319FAE2-keyring.gpg /usr/share/keyrings/
执行以下命令安装显卡驱动
apt-get update
apt-get install -y cuda-drivers
3. 使用run文件(安装CUDA)安装驱动
到CUDA Toolkit - Free Tools and Training | NVIDIA Developer下载对应版本的CUDA ,这里选择下载12.9版本。
运行下面命令,安装CUDA
wget https://developer.download.nvidia.com/compute/cuda/12.9.0/local_installers/cuda_12.9.0_575.51.03_linux.run
sudo sh cuda_12.9.0_575.51.03_linux.run
安装过程中,会出现Driver选项,选择安装即可
4. 验证驱动是否安装完成
nvidia-smi