AMD显卡(RX6750gre)+ubuntu22.04基于docker部署ROCm环境(支持CUDA)

写在前面

个人感觉,到2025年3月8日,AMD显卡的深度学习支持已经做得相当给力了.
我尝试了三种在AMD显卡上部署支持CUDA的ROCm环境的方式:

1. 使用windows操作系统上的基于docker的ubuntu20.04容器环境
2. 使用windows操作系统上的wsl2(linux子系统)环境
3. 使用ubuntu22.04操作系统上的基于docker的ubuntu22.04.5容器(使用最新的官方镜像)
   前两种完全失败,因为windows系统完全不支持AMD显卡的ROCm驱动,具体表现为
   amdgpu kfd模块无法安装.
   所以,一定要使用直接安装在物理机上的ubuntu系统.如果你的电脑上只有windows系统,建议压缩任意磁盘空间后,安装ubuntu系统.

详细系统相关信息

• CPU: AMD Ryzen5 5600
• GPU:AMD Radeon rx 6750 gre 10GB
• OS: Ubuntu 22.04.5 LTS

操作过程

第一步,在你的ubuntu系统上安装amd的ROCm驱动

官方下载地址:
https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/install-methods/amdgpu-installer/amdgpu-installer-ubuntu.html