Ubuntu上AMD显卡能够使用的stable diffusion webui部署方案流程

2022-12-02 19:31:50 浏览数 (1)

本文环境使用实体设备，显卡是AMD RX6800XT。 系统是ubuntu20.04.5 LTS 桌面版，部署成功后也可以在评论区发出你的显卡型号和系统。

非虚拟化环境。分个100G的分区安装Ubuntu，可以与win共存！想用哪个系统就重启切换！慢慢看，中间涉及git、pip安装慢的可以装个“开发者边车”解决

一、更新apt并安装一些需要的程序

代码语言：javascript复制
sudo apt-get update

sudo apt-get upgrade

sudo apt-get install wget gnupg2 git

二、安装amdgpu驱动

代码语言：javascript复制
sudo usermod -a -G video $LOGNAME

echo 'ADD_EXTRA_GROUPS=1' | sudo tee -a /etc/adduser.conf

echo 'EXTRA_GROUPS=video' | sudo tee -a /etc/adduser.conf

echo 'EXTRA_GROUPS=render' | sudo tee -a /etc/adduser.conf

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

更改amd rocm源为：https://repo.radeon.com/rocm/apt/5.1.1 （一定要改，很重要！！！）

代码语言：javascript复制
sudo amdgpu-install --usecase=hiplibsdk,rocm --no-dkms

漫长的等待...

代码语言：javascript复制
sudo usermod -a -G video $LOGNAME
sudo usermod -a -G render $LOGNAME
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，其他的我就不知道了）

代码语言：javascript复制
rocm-smi
sudo /opt/rocm/bin/rocminfo
sudo /opt/rocm/opencl/bin/clinfo

接着安装

代码语言：javascript复制
sudo apt-get install miopenkernels-gfx1030-36kdb

又是漫长的等待...

代码语言：javascript复制
sudo apt install wget git python3 python3-venv

sudo apt install python3-pip

wget https://download.pytorch.org/whl/rocm5.1.1/torch-1.12.1+rocm5.1.1-cp38-cp38-linux_x86_64.whl

sudo pip install torch-1.12.1 rocm5.1.1-cp38-cp38-linux_x86_64.whl

如果出现read timed out的话，上面的命令在来一遍

代码语言：javascript复制
sudo pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1

HSA_OVERRIDE_GFX_VERSION=10.3.0

sudo reboot     #重启电脑

进行验证

代码语言：javascript复制
python3
import torch
torch.cuda.is_available()

输出结果为 True 即可以调用gpu 然后按CTRL D退出python

现在可以下载安装本体了！！

代码语言：javascript复制
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui

cd stable-diffusion-webui

python3 -m venv venv

source venv/bin/activate

python3 -m pip install --upgrade pip wheel

这时候就去把需要的模型以及必要文件放进各个文件夹具体参考：https://www.bilibili.com/video/BV1vG411E7ea

代码语言：javascript复制
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时，说明完成安装了，然后就可以打开浏览器访问啦！下次启动只要输入：

代码语言：javascript复制
source venv/bin/activate

TORCH_COMMAND='pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1' python launch.py --precision full --no-half

剩余模型配置方面接着看：https://www.bilibili.com/video/BV1vG411E7ea

如果想在A卡上AI学习的话（未测试）请看：https://www.bilibili.com/video/BV1H8411W7VX

如果还有不懂得可以参考下：

  • https://rocmdocs.amd.com/en/latest/Installation_Guide/Installation_new.html#rocm-installation-guide-v5-0
  • https://docs.amd.com/bundle/ROCm_Installation_Guidev5.0/page/How_To_Install_ROCm.html
  • https://github.com/ROCmSoftwarePlatform/MIOpen
  • https://www.cnblogs.com/lllzhuang/articles/16083003.html
  • https://github.com/RadeonOpenCompute/ROCm/issues/1756
  • https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Install-and-Run-on-AMD-GPUs

《深入浅出MGR》视频课程

戳此小程序即可直达B站

https://www.bilibili.com/medialist/play/1363850082?business=space_collection&business_id=343928&desc=0

https 网络安全 html github git

0 人点赞