AMD显卡ubuntu系统部署comfyui

20240427更新：pytorch 2.3.0，rocm 6.0

显卡rx6600

ubuntu 20.04.6 lts

1.安装相应依赖及常用工具

sudo apt update && sudo apt install -y curl git vim ffmpeg gfortran libstdc++-12-dev cockpit openssh-server

2.在https://pytorch.org/查看ROCm版本需求，在https://repo.radeon.com/amdgpu-install/下载相应AMD官方驱动

curl -O https://repo.radeon.com/amdgpu-install/5.7.3/ubuntu/focal/amdgpu-install_5.7.50703-1_all.deb
curl -O https://repo.radeon.com/amdgpu-install/6.0.3/ubuntu/focal/amdgpu-install_6.0.60003-1_all.deb
sudo dpkg -i amdgpu-install_5.7.50703-1_all.deb
sudo dpkg -i amdgpu-install_6.0.60003-1_all.deb
sudo amdgpu-install –usecase=hiplibsdk,rocm

卸载rocm
sudo amdgpu-install –uninstall –rocmrelease=all

将用户添加到video及render组
sudo usermod -aG video $USER
sudo usermod -aG render $USER

重启

3.安装miniconda

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash ./Miniconda3-latest-Linux-x86_64.sh

禁用自动进入基础conda环境

conda config –set auto_activate_base false

查看python最新稳定版本，创建conda环境

conda create -n comfyui python=3.10.14 -y
conda activate comfyui

删除环境
conda remove -n comfyui –all

后续在conda环境下进行

4.安装pytorch，官网查看需求获取下载地址
pip3 install torch torchvision torchaudio –index-url https://download.pytorch.org/whl/rocm5.7
pip3 install torch torchvision torchaudio –index-url https://download.pytorch.org/whl/rocm6.0

5.安装comfyui

git clone https://github.com/kijai/ComfyUI-SUPIR.git

cd ~/ComfyUI
pip3 install -r requirements.txt
网络不可达时使用镜像
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

修改launch.sh
export HF_ENDPOINT=https://hf-mirror.com
export HSA_OVERRIDE_GFX_VERSION=10.3.0 //rx6600显卡使用

启动
conda activate comfyui && cd ~/ComfyUI && bash launch.sh

6.无法连接raw.githubusercontent.com

查询域名解析https://www.ip138.com/

添加解析记录
sudo vim /etc/hosts

185.199.111.133 raw.githubusercontent.com
185.199.110.133 raw.githubusercontent.com
185.199.109.133 raw.githubusercontent.com
185.199.108.133 raw.githubusercontent.com
182.43.124.6 raw.githubusercontent.com