【WSL2/ComfyUI】Radeonでもまともな速度で画像生成したい

# パッケージ類を更新しておく
sudo apt update
sudo apt upgrade -y

# AMDの手順に従ってamdgpu-installをインストールする
wget https://repo.radeon.com/amdgpu-install/6.1.3/ubuntu/jammy/amdgpu-install_6.1.60103-1_all.deb
sudo apt install ./amdgpu-install_6.1.60103-1_all.deb -y

# 手元の環境では一部パッケージが保留されたので更新しておく
sudo apt upgrade python3-update-manager update-manager-core -y

# ROCmをインストール
amdgpu-install -y --usecase=wsl,rocm,hip

...
  Name:                    gfx1100
  Marketing Name:          AMD Radeon RX 7900 XTX
  Vendor Name:             AMD
...

# pipのインストール(デフォルトでインストール済みのはず)
sudo apt install python3-pip -y

# pip、wheelの更新
pip3 install --upgrade pip wheel

# PyTorch等をインストール
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1.3/torch-2.1.2%2Brocm6.1.3-cp310-cp310-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1.3/torchvision-0.16.1%2Brocm6.1.3-cp310-cp310-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1.3/pytorch_triton_rocm-2.1.0%2Brocm6.1.3.4d510c3a44-cp310-cp310-linux_x86_64.whl
pip3 uninstall torch torchvision pytorch-triton-rocm numpy
pip3 install *.whl numpy==1.26.4

# ライブラリを更新
pushd .
location=`pip show torch | grep Location | awk -F ": " '{print $2}'`
cd ${location}/torch/lib/
rm libhsa-runtime64.so*
cp /opt/rocm/lib/libhsa-runtime64.so.1.2 libhsa-runtime64.so
popd

python3 -c 'import torch; print(torch.cuda.is_available())'

# ComfyUIをクローン
git clone https://github.com/comfyanonymous/ComfyUI.git
cd ComfyUI

# ComfyUI Managerのインストール ※オプション
git clone https://github.com/ltdrdata/ComfyUI-Manager.git ./custom_nodes/ComfyUI-Manager
pushd .
cd ./custom_nodes/ComfyUI-Manager
pip3 install -r requirements.txt --user
popd

# 必要なパッケージをインストール
pip3 install safetensors psutil einops transformers scipy torchsde aiohttp spandrel

# ComfyUIを起動
python3 main.py