今更ながらAMD GPUでStable Diffusionを動かせた

前回いろいろ試した挙句結局ダメだったのですが、今回は成功しました!

筆者の環境は前回の記事にある通りです。変わっていません。

結論

• https://github.com/AUTOMATIC1111/stable-diffusion-webui
  • stable-diffusion-webuiを使用
• https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Install-and-Run-on-AMD-GPUs
  • AMD GPU用の設定が書いてあるページ

細かい手順

AMD GPU用の設定ページ準拠なので、最新の情報は必ず上のリンクから元情報を参照してください。

手順に従いコマンドを実行

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

TORCH_COMMAND='pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1' python launch.py --precision full --no-half

詰まったところ1 – インストール

最後のコマンドでうまく行かなかったので色々調べたところ

HSA_OVERRIDE_GFX_VERSION=10.3.0 PYTHONPATH="." <最後のコマンド>

とすると良いという情報があり、これ通りに実行したら無事インストールされました。

詰まったところ2 – 画像生成

画像生成しても灰色の画像しか出ない問題が再発したのですが、起動時のオプションに `–no-half` をつけたところ解決しました。

最終的な起動コマンドは以下のとおりです。

HSA_OVERRIDE_GFX_VERSION=10.3.0 PYTHONPATH="." python3.10 launch.py --no-half

無事画像出力されました！

CPUを使ってちまちま空き時間に生成していたのですが、これからはかなりのスピードで色々試せそうです。

体感、5分 -> 30秒 となり満足のいく結果になりました。

めでたしめでたし!