AMD显卡Ubuntu20.4系统下部署stable diffusion

今天来给大家做一个AMD显卡使用stable diffusion的小白（硬核）教程。最近这段时间AI特别火，很多小伙伴看到各种大佬用AI画的老婆非常精美（色情），弄的人心痒痒，自己也想画一个心仪的老婆（画作）。

太露骨的图片不能放，你们懂得！！

但是很多小伙伴用的是AMD的独立显卡，而自从矿潮以来，显卡价格就居高不下，特别是英伟达的显卡，虽然它用AI作图有优势，但是贵的吓人，从哥布林洞窟里解救出来的RTX3070也还要2000多，简直离谱。本教程呢适用于只有一张amd的显卡，不想换英伟达显卡（穷），又想玩stable diffusion的小伙伴。

那么今天呢我用的是ubuntu 20.4.5桌面版系统部署stable diffusion的教程，保证你能顺利跑起stable diffusion，因为我是从一个完全不懂linux系统的小白找教程一步一步部署下来的，你也能行！

要部署stable diffusion的话需要满足下面三个条件：

第一，要有一个128G以上的硬盘空间，用于安装ubuntu系统。

第二，最好要有一个科学上网的环境。

第三，要有成功部署的决心，遇到什么问题呢就解决什么问题，不要随便放弃，跟着我一步一步来。

本次的部署环境呢，是AMD的独立显卡RX6600。下图呢是支持AMD显卡部署stable diffusion的型号，从470到580、vega56、64、duo、五千以及六千系列，几乎涵盖了主流的AMD显卡型号。

首先，我们需要安装Ubuntu20.4.5桌面版系统的安装我就不详细讲了，很基础，大概十分钟就能装好。为什么需要大家安装新系统呢，因为已有的ubuntu系统会因为我们装的一些软件会产生奇奇怪怪的问题。

本次部署呢，分为以下三个部分：

第一，python3.10.6的安装。

第二，安装ROCM与pytorch

第三，安装stable-diffusion

第一部分：安装python3.10.6

我们进入系统以后呢，需要安装python3.10.6，因为ubuntu20.4.5呢自带的是python3.8.10。本次安装的stable diffusion需要跑在python3.10.6环境下,当然更高版本的也能跑，只是我没试过。

1.安装依赖 如果出现选择，输入Y。

sudo -i

sudo apt-get update

sudo apt-get upgrade

sudo apt install build-essential zlib1g-dev libncurses5-dev libgdbm-dev libnss3-dev libssl-dev libreadline-dev libffi-dev libsqlite3-dev wget libbz2-dev

2.从官网下载源文件

wget https://www.python.org/ftp/python/3.10.6/Python-3.10.6.tgz

3.解压缩Python3.10文件

①先通过find命令找到Python-3.10.6.tgz的下载在哪里了：

find / -name Python-3.10.6.tgz

②接着转换路径至home：

cd /home/

③将python3.10从/root/Python-3.10.6.tgz（第①步找到的路径,红色的是我的路径）的文件复制到home文件夹下：

cp /root/Python-3.10.6.tgz /home/

④然后将文件解压至home文件夹下：tar -zvxf Python-3.10.6.tgz

tar -zvxf Python-3.10.6.tgz

4.开始安装python3.10.6

cd Python-3.10.6/

./configure --enable-optimizations

5.编译：make（没有安装make命令的安装一下）

make

6.安装sudo make install

sudo make install

这样python3.10就安装完成了。

最后，我们需要更新一下python让python默认指向为python3.10

1.更新python默认指向有以下两个方法：

（下面这两个方法是或的关系，不是且的关系！当然，你也可以都做一遍。）

①建立软连接

首先，删除原有链接

rm /usr/bin/python

其次，找到python3的安装路径（我的是在/usr/local/bin/python3）

which python3

然后，找到的路径建立新链接

ln -s /usr/local/bin/python3 /usr/bin/python

这样输入python就可以了。

②设置python的环境变量(需要搜索vim编辑器的用法)

首先，编辑.bashrc文件

vim ~/.bashrc

在这里写入自己刚刚安装python3.10的路径：

alias python='/usr/local/bin/python3'

然后：

source ~/.bashrc

python

这样就可以了！

2.更新pip默认指向

pip指向pip3也是如上的两个办法。

这里只展示方法①（如果新装的系统就直接用我的命令）：

which pip3

ln -s /usr/local/bin/pip3 /usr/bin/pip

pip

到这里，小伙伴们可以愉快的使用python3.10了！

我们回到主界面，输入 python -V 返回python 3.10.6 就说明部署成功了。

第二部分：安装ROCM与pytorch

1．ROCM的简介

ROCM和NVDIA的CUDA类似，ROCM是AMD发布的一个用于加速GPU计算的软件平台。而该平台从面世至今仅支持Linux系统，因此本文的一切操作都将在Linux系统下进行。

2.ROCm的配置

2.1 查看显卡型号

由于AMD每更新一次ROCm就抛弃一批老型号显卡支持的不做人行为，安装前确认自己的显卡是否能支持就变成了一件很重要的事。

上表包括了大部分常见的A卡型号，以世代分类。 不在表中的可以去这里输入自己显卡的型号，找到Graphics/Compute这一列会有说明。

2.2 ROCm环境的准备

2.2.1 软件包安装

首先安装AMD显卡驱动：

sudo apt-get update

wget https://repo.radeon.com/amdgpu-install/5.3/ubuntu/focal/amdgpu-install_5.3.50300-1_all.deb

sudo apt-get install ./amdgpu-install_5.3.50300-1_all.deb

执行安装操作：

sudo amdgpu-install --usecase=hiplibsdk,rocm --no-dkms

如有另外安装显卡驱动的需求，就删去后面的 --no-dkms

…漫长的等待…（大概需要一个小时左右，和网络状况关系很大）

如果顺利的话基本不会出现报错，本着warnings不用管的原则直接进入下一步即可。

真报错了原因一般是Linux内核版本不一致造成部分编译不通过，可以查看AMD的官方文档来确定所需的Linux内核版本。

查看系统内核版本：终端输入 uname -r

以楼主安装的ROCM5.3.0为例，对应的内核版本是5.15.0-46-generic，实际系统中是5.15.0-50-generic,安装时未出现报错。

2.2.2 后续配置

添加用户组：

sudo usermod -a -G video $LOGNAME

sudo usermod -a -G render $LOGNAME

配置环境变量：

echo 'export PATH=$PATH:/opt/rocm-5.3.0/bin:/opt/rocm-5.3.0/opencl/bin'

export LD_LIBRARY_PATH=/opt/rocm-5.3.0/lib:/opt/rocm-5.3.0/lib64

安装完成后重启：

sudo reboot

2.2.3 测试

打开终端输入rocminfo或rocm-smi，会出现当前设备信息和状态。

设备信息里出现GPU，且能正常显示显卡状态，即代表ROCM环境安装成功。

3 .运行库的补充及pytorch的安装

安装完成后仍需补充一些额外的运行库以免报错，测试时按照提示缺什么补什么即可，一般情况下需要补的有这些：

sudo apt install git python3 python3-pip python3-venv rocm-libs miopen-hip rccl libtinfo5 libopenmpi3 libopenblas-dev

pytorch安装，直接去官网下最新的就行

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2

4 .环境配置综合测试（本条可以不做！！！已经试过出错好像也没问题）

这里使用 https://github.com/xuhuisheng/rocm-build 提供的检查程序，可以clone或直接下载下来。

找到里面的check文件夹，终端下运行check.sh和另外3个py：

sh check.sh

python test-pytorch-device.py

python test-pytorch-fc.py

python test-pytorch-rocblas.py

运行不报错即可认为ROCm、pytorch环境配置正确。

第三部分：stable-diffusion的本地部署

把stable diffusion软件本体clone下来，参照官方文档说明部署即可。

因为已经在之前的步骤中安装了pytorch，此处可不用安装。

git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui

cd stable-diffusion-webui

python -m venv venv

source ./venv/bin/activate

python -m pip install --upgrade pip wheel

python launch.py --precision full --no-half

可能发生的错误：

错误1:torch unable to use gpu....

解决办法：

#打开launch.py找到这句代码

安装vim编辑器（已经安装过的可以不用管）

apt-get install vim

进入软件目录

cd /root/stable-diffusion-webui

编辑launch.py

vim launch.py

按“i”修改文件：

将

commandline_args = os.environ.get('COMMANDLINE_ARGS', "")

#改成

commandline_args = os.environ.get('COMMANDLINE_ARGS', "--skip-torch-cuda-test")

按ESC然后输入 :wq回车保存

错误2:No module named ‘_lzma‘

解决办法：

打开python3

cd /usr/local/lib/python3.10

安装

apt-get install liblzma-dev -y

pip3 install backports.lzma

环境中修改lzma.py位置依自己的环境路径而定，新装系统和我的路径是一样的不用改

然后

cd /usr/local/lib/python3.10

编辑

vim lzma.py

按“i”修改文件，如下的部分：

from _lzma import *

from _lzma import _encode_filter_properties, _decode_filter_properties

修改成如下：

try:

from _lzma import *

from _lzma import _encode_filter_properties, _decode_filter_properties

except ImportError:

from backports.lzma import *

from backports.lzma import _encode_filter_properties, _decode_filter_properties

按ESC然后输入 :wq回车保存

错误3:No module named 'backports'

pip3 install backports.lzma

apt-get update

apt-get upgrade

重新启动一下 stable diffusion

cd stable-diffusion-webui

source ./venv/bin/activate

python launch.py --precision full --no-half

cd /usr/local/lib/python3.10

cd ..

python

import lzma

然后就可以退出了

首次运行时会安装依赖包，所以依旧是漫长的等待…

模型文件需放置在stable-diffusion-webui/models/Stable-diffusion/ 文件夹下。可以提前放置，不放也可以，只是第一次回下载默认模型包有点慢。

跑完出现127.0.0.1:7860即代表部署成功，浏览器中输入127.0.0.1:7860可打开webUI。

stable-diffusion的运行，及常用参数

后续运行只需要输入：

cd /root/stable-diffusion-webui

HSA_OVERRIDE_GFX_VERSION=10.3.0 python launch.py --precision full --no-half

如果第一次运行显卡不工作，是CPU在跑图，则需要重启下电脑。然后取得root权限再运行，即：

sudo -i

输入密码就可以了

浏览器中输入 127.0.0.1:7860 即可打开webUI。

然后就可以愉快的玩耍啦！！！

------------------------------------------------------------------------------------------------

分割线

常用参数：

--precision full 和 --no-half：强制双精度浮点运算（据说在N卡上可以减少绿屏bug出现的几率，A卡上不知道有啥用）

--medvram 和 --lowvram：降低显存占用量，视自己显卡水平加不加。

--opt-split-attention-v1：性能优化，进一步降低显存占用量。

至于具体能降低多少，楼主之前在只有4G显存的R9 nano上运行时使用了--medvram和--opt-split-attention-v1，显存占用大概25%的样子。

--disable-safe-unpickle：加载模型前有一步unpickle(原理不明，类似于哈希校验)的过程，部分用户在这里可能报错中止，加上这个参数跳过这一步可继续运行。

详细参数表见AUTOMATIC1111的文档说明。

视频教程地址：

https://www.bilibili.com/video/BV1gs4y1H7Uq/?spm_id_from=333.999.0.0&vd_source=798c720594d31bfcbf5d2a56af4707f4

https://www.youtube.com/watch?v=FLOtfHvX8NE