在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
© 版权声明
THE END
若本文对您有帮助,欢迎点赞打赏转发
您的支持将是作者更新最大的动力
点赞49打赏 分享
评论 抢沙发

请登录后发表评论

    暂无评论内容