Ubuntu 22.04+RTX 5060:深度学习环境搭建之驱动先行方案
·
Ubuntu 22.04 + RTX 5060:深度学习驱动安装指南
一、关键准备
-
禁用默认驱动
sudo nano /etc/modprobe.d/blacklist-nouveau.conf添加以下内容:
blacklist nouveau options nouveau modeset=0更新并重启:
sudo update-initramfs -u sudo reboot -
验证禁用状态
lsmod | grep nouveau # 若无输出则成功
二、驱动安装流程
-
添加官方驱动源
sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt update -
查找适配驱动(RTX 5060需≥535版本)
ubuntu-drivers devices输出示例:
driver: nvidia-driver-535 - distro non-free recommended -
安装推荐驱动
sudo apt install nvidia-driver-535
三、安装后验证
-
基础功能检查
nvidia-smi # 应显示GPU信息成功输出示例:
+---------------------------------------------------------------------------------------+ | NVIDIA-SMI 535.86.05 Driver Version: 535.86.05 CUDA Version: 12.2 | |-----------------------------------------+----------------------+----------------------+ | GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. | |=========================================+======================+======================| | 0 NVIDIA GeForce RTX 5060 On | 00000000:01:00.0 On | N/A | -
OpenGL兼容性测试
glxinfo | grep "OpenGL renderer" # 应显示NVIDIA GPU型号
四、深度学习环境准备
-
安装CUDA Toolkit
sudo apt install nvidia-cuda-toolkit nvcc --version # 验证CUDA编译器 -
配置环境变量
echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc source ~/.bashrc
五、疑难排解
| 问题现象 | 解决方案 |
|---|---|
| 启动黑屏 | 进入恢复模式卸载驱动:sudo apt purge nvidia* |
| CUDA版本不匹配 | 手动下载CUDA Toolkit:https://developer.nvidia.com/cuda-toolkit |
| GPU未识别 | 检查PCIe连接:`lspci |
关键提示:
- 首次安装后需完全重启系统:
sudo shutdown -r now- 推荐使用长期支持版本驱动(标记为LTSB)
- 避免混合安装不同来源的驱动包
完成驱动安装后,可通过以下命令监控GPU状态:
watch -n 1 nvidia-smi # 实时刷新GPU使用率
更多推荐
所有评论(0)