1. 为什么需要专门配置AMD显卡驱动很多刚接触Ubuntu深度学习的新手可能会有疑问Linux内核不是已经自带AMD开源驱动了吗确实默认的amdgpu驱动能支持基础显示功能但当你需要运行PyTorch等深度学习框架时就会遇到三个致命问题首先开源驱动缺乏对ROCm计算平台的支持。我实测过RX 6600显卡使用默认驱动时torch.cuda.is_available()永远返回False。其次开源驱动无法发挥显卡的全部性能在ResNet50模型训练中专有驱动的速度能快3倍以上。最后像RX 6000系列等新显卡的某些功能如光线追踪单元需要专有驱动才能启用。提示在开始前建议运行lspci -k | grep -A 2 VGA确认显卡型号RX 5000/6000系列用户强烈建议继续阅读后续配置步骤。2. 驱动安装前的准备工作2.1 系统环境检查我的测试环境是Ubuntu 20.04.6 LTS但该方法同样适用于22.04。关键要确保内核版本不低于5.4可以通过uname -r查看。如果内核太旧先运行sudo apt update sudo apt upgrade -y2.2 卸载旧驱动如果你之前折腾过驱动安装比如用过第三方PPA强烈建议先彻底清理sudo apt purge amdgpu* rocm* sudo reboot这个步骤我踩过坑——残留的旧驱动文件会导致后续安装失败特别是/etc/apt/sources.list.d/目录下的残留配置。3. 一站式安装AMD专有驱动与ROCm3.1 添加官方仓库安装这是AMD推荐的标准方法适合大多数用户wget https://repo.radeon.com/amdgpu-install/22.20/ubuntu/focal/amdgpu-install_22.20.50200-1_all.deb sudo apt install ./amdgpu-install_22.20.50200-1_all.deb sudo amdgpu-install --usecasegraphics,opencl,rocm --no-dkms这里有几个关键参数需要注意--no-dkms避免内核模块编译问题Secure Boot用户需移除该参数--usecase指定安装组件深度学习必须包含rocm3.2 手动安装特定版本当需要特定ROCm版本时比如PyTorch官方要求的版本可以这样操作sudo apt install rocm-hip-sdk5.3.3 rocm-opencl-runtime5.3.3安装完成后务必验证设备权限ls -l /dev/dri/render* sudo usermod -a -G render,video $USER4. 深度学习环境验证与调优4.1 基础功能验证重启后运行这三个命令检查安装状态rocm-smi # 应显示GPU功耗和温度 /opt/rocm/bin/rocminfo # 检查计算设备信息 clinfo | grep Device Name # 确认OpenCL正确识别4.2 PyTorch环境配置现在安装支持ROCm的PyTorchpip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.4.2验证GPU是否可用import torch print(torch.cuda.is_available()) # 应输出True print(torch.rand(10,10).to(cuda)) # 测试张量计算4.3 性能调优技巧在/etc/default/grub中添加这些参数可提升性能GRUB_CMDLINE_LINUXamdgpu.ppfeaturemask0xffffffff amdgpu.vm_fragment_size9更新grub后重启sudo update-grub sudo reboot5. 常见问题解决方案5.1 黑屏问题处理如果安装后出现黑屏尝试在GRUB界面按e键修改启动参数在linux行末尾添加amdgpu.runpm0 amdgpu.dc0进入系统后重新安装驱动并启用DKMSsudo amdgpu-install --usecasegraphics,rocm --dkms5.2 ROCm检测失败当rocminfo报错时通常是权限问题导致sudo chmod 666 /dev/kfd sudo chmod 666 /dev/dri/render*如果问题依旧检查内核日志dmesg | grep -i amdgpu5.3 PyTorch版本兼容性不同PyTorch版本需要特定ROCm版本支持这是实测可用的组合表PyTorch版本ROCm版本备注2.05.4.2推荐组合1.125.3适合旧代码迁移1.84.5仅限老旧系统6. 进阶配置指南对于需要多GPU训练的用户还需要配置HCC和RCClsudo apt install hcc rocminfo rccl export HCC_AMDGPU_TARGETgfx1030 # RX6000系列设为gfx1030在~/.bashrc中添加这些环境变量能提升性能export HSA_OVERRIDE_GFX_VERSION10.3.0 export ROCBLAS_TENSILE_LIBPATH/opt/rocm/lib/rocblas/library