首先下载驱动程序的安装脚本。目前,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是否成功安装。如果返回错误,则安装过程中出现了问题。