📌 1. 安装前准备 🧰
在开始安装前,请确保你的系统满足以下要求:
- 操作系统:Ubuntu 20.04 / CentOS 8 / Windows 10(支持NVIDIA显卡)
- 显卡:NVIDIA GPU(如TITAN X、RTX 3090等)
- 驱动:已安装最新版NVIDIA驱动(推荐使用
nvidia-smi
检查版本)
📦 2. 安装步骤 📚
✅ 步骤一:安装NVIDIA驱动
访问NVIDIA官方驱动下载页面选择对应系统的驱动版本,使用以下命令安装:
sudo apt update
sudo apt install nvidia-driver-535
📌 安装完成后重启系统:sudo reboot
✅ 步骤二:安装CUDA Toolkit
前往官方CUDA安装指南获取详细说明,推荐使用以下命令安装:
sudo apt install cuda-toolkit-11-8
💡 安装过程中会自动配置环境变量,可检查~/.bashrc
文件
✅ 步骤三:验证安装
运行测试程序确认安装成功:
nvcc --version
nvidia-smi
✅ 正常输出显示版本号即为安装成功
📈 3. 性能优化建议 📈
- 使用
nvidia-smi --query-gpu=temperature.gpu,utilization.gpu --format=csv
监控GPU状态 - 安装cuDNN库提升深度学习性能:cuDNN安装教程
- 配置CUDA环境变量:
export PATH=/usr/local/cuda-11.8/bin:$PATH
🌐 4. 常见问题排查 🔍
问题 | 解决方案 |
---|---|
nvcc: command not found |
检查环境变量是否配置正确 |
安装失败 | 尝试使用sudo apt --fix-broken install 修复 |
驱动冲突 | 使用sudo nvidia-uninstall 卸载旧版本 |