「亲测」在linux系统安装rx580/rx570显卡驱动支持Pytorch2实现AI绘画提速

构建带有 ROCm 支持的 pytorch 2.1.2，以实现在Linux下WebUI及ComfyUI加速AI绘画，亲测可用。

系统说明

Linux Mint 21.2
Radeon RX 580 8GB
RoCm 5.5

Python 3.10.6
- pytorch 2.1.2
- torchvision 0.16.2

本教程适合RX580/RX570，系统采用Linux Mint，安装的Rocm、Python、Pytorch、Torchvision版本如上。

安装步骤

安装依赖项

sudo apt autoremove rocm-core amdgpu-dkms
sudo apt install libopenmpi3 libstdc++-12-dev libdnnl-dev ninja-build libopenblas-dev libpng-dev libjpeg-dev

安装 ROCm

sudo -i
sudo echo ROC_ENABLE_PRE_VEGA=1 >> /etc/environment
sudo echo HSA_OVERRIDE_GFX_VERSION=8.0.3 >> /etc/environment
# 完成此步后重启

wget https://repo.radeon.com/amdgpu-install/5.5/ubuntu/jammy/amdgpu-install_5.5.50500-1_all.deb
sudo apt install ./amdgpu-install_5.5.50500-1_all.deb
sudo amdgpu-install -y --usecase=rocm,hiplibsdk,mlsdk

sudo usermod -aG video $LOGNAME
sudo usermod -aG render $LOGNAME

# 验证
rocminfo
clinfo

# 以上命令输出对应信息成功

构建torch及torchvison

您可能需要安装额外的依赖项，并且构建将需要很长时间。

首先在主目录创建目录pytorch2.1.2，然后按以下命令构建pytorch：

cd pytorch2.1.2
git clone --recursive https://github.com/pytorch/pytorch.git -b v2.1.2
cd pytorch
pip install cmake mkl mkl-include
pip install -r requirements.txt
sudo ln -s /usr/lib/x86_64-linux-gnu/librt.so.1 /usr/lib/x86_64-linux-gnu/librt.so
export PATH=/opt/rocm/bin:$PATH ROCM_PATH=/opt/rocm HIP_PATH=/opt/rocm/hip
export PYTORCH_ROCM_ARCH=gfx803
export PYTORCH_BUILD_VERSION=2.1.2 PYTORCH_BUILD_NUMBER=1
export USE_CUDA=0 USE_ROCM=1 USE_NINJA=1
python3 tools/amd_build/build_amd.py
python3 setup.py bdist_wheel

然后构建pytorchvision

cd ..
git clone https://github.com/pytorch/vision.git -b v0.16.2
cd vision
export BUILD_VERSION=0.16.2
FORCE_CUDA=1 ROCM_HOME=/opt/rocm/ python3 setup.py bdist_wheel

如果不想自自己构建，也可下载已构建好的版本：

torch-2.1.2-cp310-cp310-linux_x86_64.whl

torchvision-0.16.2-cp310-cp310-linux_x86_64.whl

注意：以上文件需要能访问github。

安装

可安装在python虚拟环境，也可安装在主环境，这里安装在主环境。

sudo ln -s /usr/bin/python3.10 /usr/bin/python
python -m pip install --upgrade pip wheel
pip uninstall torch torchvision
pip install /home/用户名pytorch2.1.2/pytorch/dist/torch-2.1.2-cp310-cp310-linux_x86_64.whl
pip install /home/用户名/pytorch2.1.2/vision/dist/torchvision-0.16.2-cp310-cp310-linux_x86_64.whl
pip list | grep 'torch'

请将以上的“用户名”替换为你的主目录用户名

测试

python
import torch
torch.cuda.is_available()

WebUI安装

python -m venv venv --system-site-packages
source venv/bin/activate
pip install -r requirements.txt
python launch.py