Ubuntu 22.04 + RTX 5060:深度学习驱动安装指南

一、关键准备
  1. 禁用默认驱动

    sudo nano /etc/modprobe.d/blacklist-nouveau.conf
    

    添加以下内容:

    blacklist nouveau
    options nouveau modeset=0
    

    更新并重启:

    sudo update-initramfs -u
    sudo reboot
    

  2. 验证禁用状态

    lsmod | grep nouveau  # 若无输出则成功
    

二、驱动安装流程
  1. 添加官方驱动源

    sudo add-apt-repository ppa:graphics-drivers/ppa
    sudo apt update
    

  2. 查找适配驱动(RTX 5060需≥535版本)

    ubuntu-drivers devices
    

    输出示例:

    driver: nvidia-driver-535 - distro non-free recommended
    

  3. 安装推荐驱动

    sudo apt install nvidia-driver-535
    

三、安装后验证
  1. 基础功能检查

    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 |
    

  2. OpenGL兼容性测试

    glxinfo | grep "OpenGL renderer"  # 应显示NVIDIA GPU型号
    

四、深度学习环境准备
  1. 安装CUDA Toolkit

    sudo apt install nvidia-cuda-toolkit
    nvcc --version  # 验证CUDA编译器
    

  2. 配置环境变量

    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

关键提示

  1. 首次安装后需完全重启系统sudo shutdown -r now
  2. 推荐使用长期支持版本驱动(标记为LTSB)
  3. 避免混合安装不同来源的驱动包

完成驱动安装后,可通过以下命令监控GPU状态:

watch -n 1 nvidia-smi  # 实时刷新GPU使用率

更多推荐