共计 2176 个字符,预计需要花费 6 分钟才能阅读完成。
背景介绍
NVIDIA 算力卡驱动是 GPU 加速计算和 AI 开发的基石。无论是运行 TensorFlow、PyTorch 等深度学习框架,还是进行高性能计算(HPC),正确安装和配置驱动都是第一步。CentOS8 作为企业级 Linux 发行版,其稳定性和安全性使其成为许多计算场景的首选,但官方 NVIDIA 驱动安装过程可能会遇到内核版本不匹配、驱动冲突等问题,导致安装失败或性能不佳。

准备工作
在开始安装前,需要确保系统环境满足基本要求,并准备好必要的工具和依赖。
- 系统要求检查
- 确认系统版本:
cat /etc/redhat-release应显示 CentOS 8.x - 确认 GPU 型号:
lspci | grep -i nvidia应列出你的 NVIDIA GPU -
确保系统已更新:
sudo dnf update -y -
安装内核开发工具
- 安装必要的开发工具和内核头文件:
sudo dnf install -y gcc make kernel-devel kernel-headers - 确认内核版本匹配:
uname -r应与/usr/src/kernels下的目录一致
分步安装指南
禁用 nouveau 驱动
Nouveau 是 Linux 自带的 NVIDIA 开源驱动,可能与官方驱动冲突,需要先禁用。
-
创建禁用配置文件:
sudo bash -c "echo'blacklist nouveau'>> /etc/modprobe.d/blacklist-nouveau.conf" sudo bash -c "echo'options nouveau modeset=0'>> /etc/modprobe.d/blacklist-nouveau.conf" -
重建 initramfs 并重启:
sudo dracut --force sudo reboot -
验证 nouveau 是否已禁用:
lsmod | grep nouveau如果没有输出,则表示禁用成功。
下载并安装 NVIDIA 驱动
- 选择驱动版本
- 访问 NVIDIA 官方驱动下载页面(https://www.nvidia.com/Download/index.aspx)
-
根据你的 GPU 型号和 CUDA 需求选择合适的驱动版本
-
下载驱动
wget https://us.download.nvidia.com/XFree86/Linux-x86_64/< 驱动版本 >/NVIDIA-Linux-x86_64-< 驱动版本 >.run -
安装驱动
chmod +x NVIDIA-Linux-x86_64-< 驱动版本 >.run sudo ./NVIDIA-Linux-x86_64-< 驱动版本 >.run安装过程中可能会提示以下选项:
- 安装 32 位兼容库:根据需求选择
- 自动更新 X 配置:建议选择 Yes
-
安装 DKMS 模块:建议选择 Yes(便于内核更新后自动重建驱动)
-
验证驱动安装
nvidia-smi如果看到 GPU 信息和驱动版本,则安装成功。
CUDA 环境配置
虽然驱动已经安装,但为了完整的 GPU 计算能力,通常还需要安装 CUDA 工具包。
- 下载 CUDA 工具包
- 访问 NVIDIA CUDA 下载页面(https://developer.nvidia.com/cuda-downloads)
- 选择与驱动兼容的 CUDA 版本(注意版本匹配)
-
选择 CentOS8 的 rpm(network)安装方式
-
安装 CUDA
sudo dnf module install nvidia-driver:latest-dkms sudo dnf install cuda -
配置环境变量
在~/.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 版本信息。
常见问题排查
- 驱动安装失败:内核模块无法编译
- 确保安装了正确版本的内核头文件:
sudo dnf install kernel-devel-$(uname -r) -
如果内核已更新但未重启,请重启后重试
-
X 服务器已运行导致安装失败
- 切换到文本模式:
sudo systemctl isolate multi-user.target -
安装完成后再切换回图形模式:
sudo systemctl isolate graphical.target -
nvidia-smi 命令找不到
- 检查驱动是否安装成功
- 确保 /usr/bin 在 PATH 中
-
尝试重新安装驱动
-
CUDA 版本与驱动不兼容
- 参考 NVIDIA 官方文档检查版本兼容性
- 卸载不兼容版本并安装正确版本
最佳实践
- 驱动维护策略
- 定期检查驱动更新,但不要盲目升级
- 在生产环境中,建议使用长期支持(LTS)版本的驱动
-
考虑使用 DKMS 来自动处理内核更新后的驱动重建
-
环境隔离
- 对于不同的 AI 框架,建议使用 conda 或 Docker 创建隔离环境
-
这样可以在不同项目中使用不同的 CUDA 版本
-
监控与日志
- 定期检查
nvidia-smi输出,监控 GPU 使用情况 - 驱动日志位于 /var/log/nvidia-installer.log
结语
通过以上步骤,你应该已经在 CentOS8 系统上成功安装了 NVIDIA 算力卡驱动并配置了 CUDA 环境。现在,你可以开始探索 GPU 加速的计算世界了!建议尝试在自己的 GPU 型号上复现这个安装过程,并记录下任何遇到的特殊问题及其解决方案,这对未来维护会很有帮助。
