构建带有 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
暂无评论内容