Ubuntu22.04.2 Lts安装ComfyUI，实现RX580双显卡AI绘画

由于我是AMD显卡，在Win下无法实现AI绘画GPU加速，虽然安装了Macos苹果系统，能利用MPS进行加速，但还是不太满意。于是，便准备在Linux系统下利用rocm进行AI绘画，看是否在速度上有所提升。

由于我是RX580显卡，比较老旧，新版rocm已停止对该卡支持。经过一翻查阅资料，并实际摸索，最后终于成功安装。

安装环境

Ubuntu 22.04.2 LTS
Radeon RX 580 8GB
RoCm 5.4.3
Gcc 11.4.0
Linux 5.19(内核)

Python 3.10.12
- pytorch 1.13.1
- torchvision 0.14.1

Ubuntu22.04.2安装非常简单，按官方教程制作好安装盘，直接安装即可。系统安装完成后，第一步必须改内核，6系内核经我个人测试，无法安装rocm。

更换5.19内核

查看内核：

uname -r

如果不是5.19，首先下载内核:

https://www.123pan.com/s/IwGkjv-ZKjXH.html 提取码:2XBk

也可到官方下载: https://kernel.ubuntu.com/~kernel-ppa/mainline/

安装*.deb：

sudo dpkg -i *.deb

更新grub：

sudo update-grub

安装mainline管理内核

sudo add-apt-repository ppa:cappelikan/ppa
sudo apt update
sudo apt install mainline

打开mainline，删除未使用内核。如果存在依赖无法删除指定内核，则删除相应依赖即可。

安装ROCM

sudo echo ROC_ENABLE_PRE_VEGA=1 >> /etc/environment
sudo echo HSA_OVERRIDE_GFX_VERSION=8.0.3 >> /etc/environment
# reboot

wget https://repo.radeon.com/amdgpu-install/22.40.3/ubuntu/focal/amdgpu-install_5.4.50403-1_all.deb
sudo apt install ./amdgpu-install_5.4.50403-1_all.deb
sudo amdgpu-install -y --usecase=rocm,hiplibsdk,mlsdk

sudo usermod -aG video $LOGNAME
sudo usermod -aG render $LOGNAME

# verify
rocminfo
clinfo

安装完成后重启，如果出现错误，请根据提示解决。正常情况下，不会出错。

安装pytorch及torchvision

首先安装：

sudo apt install libopenblas-dev

首先下载已经编译好的版本，然后按命令安装。

https://www.123pan.com/s/IwGkjv-eKjXH.html 提取码:YwwS

sudo apt install libopenmpi3 libstdc++-12-dev
pip install torch-1.13.1-cp310-cp310-linux_x86_64.whl
pip install torchvision-0.14.1-cp310-cp310-linux_x86_64.whl

测试是否成功

sudo ln -s /usr/bin/python3.10 /usr/bin/python
python
import torch
torch.cuda.is_available()

如果返回True表示安装成功，否则请重新安装并核对安装过程是否出错。

安装comfyui

按comfyui官方说明安装即可

启动参数

python main.py  --use-split-cross-attention

默认使用0号显卡，加上下面参数可选择显卡(仅针对多显卡)：

--cuda-device 1

总结

ubuntu安装请使用最小安装，保证系统纯净，否则可能导致失败。

经测试，跑515＊515的图，能达到2.5it/s，768*512的图能到1.1s/it，速度提升明显。