引言：本文将详细介绍在较老的 CentOS 8.5.2111 服务器上安装 NVIDIA 显卡驱动和 CUDA 工具包的步骤。
对于 Ubuntu 用户，可将 yum 替换为 apt 进行安装。
安装前请务必检查 PyTorch 和 CUDA 的版本对应关系，以避免重复安装。可以在 PyTorch 官网查看 PyTorch 与 CUDA 的对应关系，而 CUDA 和 NVIDIA 驱动版本的对应关系则可以在 CUDA 12.6 Update 2 Release Notes 中找到。
安装顺序为：显卡驱动 → CUDA → CUDA Toolkit → cuDNN → PyTorch。
以服务器上的 Tesla V100 PCIe 32GB 显卡为例，适配的 PyTorch 版本与 CUDA 12.4 对应，因此我们安装 CUDA 12.4.1 及其对应的 NVIDIA 驱动 550.54.15。
首先，检查显卡型号：
lspci | grep -i nvidia
如果显卡支持 CUDA 加速，您将看到类似 NVIDIA GeForce GTX 1080 的型号。
接下来，准备安装 NVIDIA 驱动的环境：
sudo yum -y install kernel-devel sudo yum -y install epel-release sudo yum -y install gcc
如果在 CentOS 上找不到 nvidia-driver-latest-dkms，可以尝试手动安装 NVIDIA 驱动和 CUDA。
手动下载并安装 NVIDIA 驱动：
访问 NVIDIA 官方网站下载驱动，选择您的显卡型号和操作系统。下载完成后，进入下载目录，赋予安装包执行权限并运行：
wget https://us.download.nvidia.com/tesla/550.54.15/NVIDIA-Linux-x86_64-550.54.15.run sudo chmod +x NVIDIA-Linux-x86_64-*.run sudo ./NVIDIA-Linux-x86_64-*.run
在安装过程中选择默认选项，并确保禁用 nouveau 驱动（如有提示）。安装完成后，重启系统：
sudo reboot
验证 NVIDIA 驱动是否已成功安装：
nvidia-smi
若显示 NVIDIA 驱动信息，则表明安装成功。
接下来，安装 CUDA：
添加 CUDA 存储库：
sudo yum-config-manager --add-repo=https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo
安装 CUDA 工具包：
sudo yum install -y libXi-devel libXmu-devel libXt-devel libXext-devel libX11-devel gcc-c++ wget https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run sudo sh cuda_12.4.1_550.54.15_linux.run
如果遇到临时文件空间不足的问题，可以通过设置 TMPDIR 环境变量来解决：
sudo mkdir /dev/sda1/home/tmp sudo mount --bind /dev/sda1/home/tmp /tmp sudo sh cuda_12.4.1_550.54.15_linux.run sudo umount /tmp
配置环境变量：
ls /usr/local/cuda-* echo 'export PATH=/usr/local/cuda-12.4/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda-12.4/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc source ~/.bashrc
验证 CUDA 安装：
nvcc -V
如果正确显示 CUDA 版本信息，则表明安装成功。
结论：经过一番努力，借助网络文章和官方文档的指导，最终成功安装了 NVIDIA 驱动和 CUDA。
附注：在使用阿里云镜像时遇到问题，建议切换到清华大学镜像源。将 /etc/yum.repos.d/CentOS-Base.repo 文件中的 http://mirrors.aliyun.com 替换为 https://mirrors.tuna.tsinghua.edu.cn：
sed -e "s|^mirrorlist=|#mirrorlist=|g" \
-e "s|^#baseurl=http://mirror.centos.org/centos/\$releasever|baseurl=https://mirrors.tuna.tsinghua.edu.cn/centos-vault/8.5.2111|g" \
-e "s|^#baseurl=http://mirror.centos.org/\$contentdir/\$releasever|baseurl=https://mirrors.tuna.tsinghua.edu.cn/centos-vault/8.5.2111|g" \
-i.bak \
/etc/yum.repos.d/CentOS-*.repo更新缓存并安装 EPEL 仓库：
sudo yum makecache sudo yum install epel-release
如果原始仓库文件损坏，可以从阿里云镜像站下载：
repo: https://www.php.cn/link/8857f91319bfa23d6fa43b632b4e1163
epel: https://www.php.cn/link/cbc32b0f8a746c87884c5bb10a1a4289
以上就是安装 nvidia 显卡驱动及故障排查的详细内容，更多请关注php中文网其它相关文章！
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号
827
收藏
