jetsung 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" # 应识别到显卡
jetsung Pytorch pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2