AMD显卡Rocm教程
一、更新apt并安装一些需要的程序
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install wget gnupg2 git python3 python3-venv python3-pip
二` 安装amdgpu驱动以及rocm
wget https://repo.radeon.com/amdgpu-install/22.20/ubuntu/focal/amdgpu-install_22.20.50200-1_all.deb
sudo apt-get install ./amdgpu-install_22.20.50200-1_all.deb
sudo amdgpu-install --usecase=hiplibsdk,rocm --no-dkms
三`添加系统到路径
echo 'export PATH=$PATH:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin' | sudo tee -a /etc/profile.d/rocm.sh
此时可以通过下面命令确认安装(6800xt会显示gfx1030,其他的我就不知道了)
rocm-smi
sudo /opt/rocm/bin/rocminfo
sudo /opt/rocm/opencl/bin/clinfo
现在可以下载安装本体了!!
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
cd stable-diffusion-webui
python3 -m venv venv
source venv/bin/activate
export HSA_OVERRIDE_GFX_VERSION=10.3.0
python3 -m pip install --upgrade pip wheel
TORCH_COMMAND='pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1' python launch.py --precision full --no-half
直到出现127.0.0.1:7860时,说明完成
然后就可以打开浏览器访问啦!
下次启动只要输入
source venv/bin/activate
export HSA_OVERRIDE_GFX_VERSION=10.3.0(非原生gfx1030输入命令)
python launch.py --precision full --no-half