AMD GPU 驱动安装

Radeon™ RX 580 Drivers & Support

https://www.amd.com/en/support/downloads/drivers.html/graphics/radeon-600-500-400/radeon-rx-500-series/radeon-rx-580.html

ROCm installation for Linux

https://rocm.docs.amd.com/projects/install-on-linux/en/latest/index.html

# 安装驱动
sudo amdgpu-install  

# 卸载驱动
sudo amdgpu-uninstall  

# 安装驱动
sudo apt install mesa-utils mesa-utils-extra

# 查看驱动
glxinfo | grep "OpenGL renderer"  # 应显示 "AMD Radeon RX 580"
vulkaninfo | grep "GPU id"        # 应识别到显卡

Pytorch

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2