Ubuntu上AMD显卡能够使用的stable diffusion webui部署方案流程
Posted 老叶茶馆_
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Ubuntu上AMD显卡能够使用的stable diffusion webui部署方案流程相关的知识，希望对你有一定的参考价值。
本文环境使用实体设备，显卡是AMD RX6800XT。
系统是ubuntu20.04.5 LTS 桌面版，部署成功后也可以在评论区发出你的显卡型号和系统。
非虚拟化环境。分个100G的分区安装Ubuntu，可以与win共存！想用哪个系统就重启切换！慢慢看，中间涉及git、pip安装慢的可以装个“开发者边车”解决
一、更新apt并安装一些需要的程序
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install wget gnupg2 git二、安装amdgpu驱动
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 （一定要改，很重要！！！）
sudo amdgpu-install --usecase=hiplibsdk,rocm --no-dkms漫长的等待...
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，其他的我就不知道了）
rocm-smi
sudo /opt/rocm/bin/rocminfo
sudo /opt/rocm/opencl/bin/clinfo接着安装
sudo apt-get install miopenkernels-gfx1030-36kdb又是漫长的等待...
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%2Brocm5.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的话，上面的命令在来一遍
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 #重启电脑进行验证
python3
import torch
torch.cuda.is_available()输出结果为 True 即可以调用gpu 然后按CTRL+D退出python
现在可以下载安装本体了！！
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
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
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
文章推荐：
想看更多技术好文，点个“在看”吧！
以上是关于Ubuntu上AMD显卡能够使用的stable diffusion webui部署方案流程的主要内容，如果未能解决你的问题，请参考以下文章
Ubuntu22.04.1 LTS系统上实现KVM虚拟机显卡直通（AMD/NVIDIA+板载显卡）