共计 1974 个字符,预计需要花费 5 分钟才能阅读完成。
背景介绍
在 AI 开发和 GPU 加速计算领域,NVIDIA 显卡驱动是基础且关键的组件。无论是深度学习训练、科学计算还是图形渲染,都需要正确安装和配置显卡驱动才能充分发挥 GPU 的算力。对于 CentOS8 这样的企业级 Linux 发行版,安装 NVIDIA 驱动可能比在 Ubuntu 等桌面系统上更复杂,但通过本指南,即使是新手也能顺利完成安装。

准备工作
在开始安装之前,我们需要确保系统满足基本要求,并安装必要的开发工具。
-
检查系统信息
运行以下命令确认系统版本和内核信息:cat /etc/redhat-release # 查看 CentOS 版本 uname -r # 查看内核版本 -
安装开发工具和内核头文件
这些是编译 NVIDIA 驱动所必需的:dnf groupinstall "Development Tools" -y dnf install kernel-devel-$(uname -r) kernel-headers-$(uname -r) -y -
确认 GPU 型号
使用以下命令查看系统中的 NVIDIA GPU 信息:lspci | grep -i nvidia记录下你的 GPU 型号,这将帮助你在 NVIDIA 官网下载正确的驱动版本。
分步安装指南
1. 禁用 nouveau 驱动
Nouveau 是 Linux 自带的 NVIDIA 显卡开源驱动,它会与官方驱动冲突,必须先禁用。
-
创建配置文件
echo "blacklist nouveau" >> /etc/modprobe.d/blacklist-nouveau.conf echo "options nouveau modeset=0" >> /etc/modprobe.d/blacklist-nouveau.conf -
重建 initramfs
dracut --force -
重启系统
reboot -
验证 nouveau 是否已禁用
重启后运行:lsmod | grep nouveau如果没有输出,说明禁用成功。
2. 下载 NVIDIA 驱动
-
访问 NVIDIA 官方驱动下载页面
根据你的 GPU 型号和系统信息选择合适的驱动版本。 -
下载驱动包
例如,对于 Tesla T4 显卡,可以运行:wget https://us.download.nvidia.com/tesla/450.80.02/NVIDIA-Linux-x86_64-450.80.02.run
3. 安装 NVIDIA 驱动
-
赋予执行权限
chmod +x NVIDIA-Linux-x86_64-*.run -
运行安装程序
./NVIDIA-Linux-x86_64-*.run安装过程中可能会提示一些警告,通常可以安全忽略。
-
验证安装
安装完成后,运行:nvidia-smi如果看到 GPU 信息输出,说明驱动安装成功。
CUDA 环境配置
-
下载 CUDA Toolkit
访问 NVIDIA CUDA 下载页面,选择与驱动兼容的版本。 -
安装 CUDA
wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run sh cuda_11.0.3_450.51.06_linux.run -
配置环境变量
将以下内容添加到~/.bashrc 文件中:export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH然后使配置生效:
source ~/.bashrc -
验证 CUDA 安装
nvcc --version应该能看到 CUDA 版本信息。
常见问题排查
-
内核版本不匹配
如果安装时提示内核版本问题,确保已安装与当前运行内核匹配的内核头文件:dnf install kernel-devel-$(uname -r) -
驱动安装失败
如果驱动安装失败,尝试在安全模式下安装:./NVIDIA-Linux-x86_64-*.run --no-opengl-files -
X 服务器冲突
如果系统运行了 X 服务器,可能需要先停止它:systemctl stop gdm
最佳实践
-
定期更新驱动
定期检查并更新到 NVIDIA 官方推荐的最新稳定版驱动。 -
保持内核和驱动同步
系统内核更新后,记得重新安装 NVIDIA 驱动。 -
使用 DKMS
考虑使用 DKMS 自动重建内核模块:./NVIDIA-Linux-x86_64-*.run --dkms -
生产环境建议
在生产环境中,建议: - 使用长期支持 (LTS) 驱动版本
- 在测试环境验证新驱动后再部署到生产
- 定期检查 GPU 健康状况(nvidia-smi)
结语
通过本指南,你应该已经成功在 CentOS8 系统上安装了 NVIDIA 显卡驱动并配置了 CUDA 环境。现在,你可以尝试在自己的 GPU 型号上复现这个过程,开始你的 GPU 加速计算之旅。如果在安装过程中遇到任何问题,欢迎在评论区分享,我们可以一起探讨解决方案。
