首先下载驱动程序的安装脚本。目前，PyTorch与v5.4.2兼容，因此我们将下载该版本。下面提供必要的命令，但如果需要更详细的说明，请参考AMD的页面《如何安装ROCm》。
Ubuntu v20.04
sudo apt-get update
wget https://repo.radeon.com/amdgpu-install/5.4.2/ubuntu/focal/amdgpu-install_5.4.50402-1_all.deb
sudo apt-get install ./amdgpu-install_5.4.50402-1_all.deb
Ubuntu v22.04
sudo apt-get update
wget https://repo.radeon.com/amdgpu-install/5.4.2/ubuntu/jammy/amdgpu-install_5.4.50402-1_all.deb
sudo apt-get install ./amdgpu-install_5.4.50402-1_all.deb
接下来运行安装程序脚本以安装驱动程序。--usecase下唯一需要的参数是rocm，但我建议还安装图形驱动程序、多媒体和其他一些东西，可以使用以下命令完成：
sudo amdgpu-install --usecase=dkms,graphics,multimedia,opencl,hip,hiplibsdk,rocm
其他要求
现在安装Nvidia CUDA工具包，这是运行PyTorch所需的。
sudo apt install nvidia-cuda-toolkit
现在将您的用户添加到render组（将username替换为您自己的用户名）。
sudo adduser username render
您可能还需要将用户添加到video组，尽管对我来说这一步骤不是必需的：
sudo adduser username video
下面的步骤仅适用于某些消费级GPU。如果您使用的是专业卡或具有16GB VRAM的RDNA 2 GPU（例如RX 6800 XT、6900 XT），则不需要执行以下步骤。我正在使用的较低级别的显卡，如RX 6600 XT和6700 XT，则需要执行以下步骤。旧一代的GPU可能需要其他未在此处列出的步骤，因此建议阅读Reddit主题以获取有关运行早于RDNA 2的显卡的更多信息。Reddit: HOW-TO: Stable Diffusion on an AMD GPU.
使用您选择的编辑器编辑~/.profile文件。
sudo nano ~/.profile
将以下行粘贴到文件底部，然后按ctrl-x保存文件并退出。
export HSA_OVERRIDE_GFX_VERSION=10.3.0
现在，在继续之前，请确保重新启动计算机。
然后，您可以通过运行rocm-smi或rocminfo来检查ROCm是否成功安装。如果返回错误，则安装过程中出现了问题。
