DockerでStable Diffusionを動かす(GPU版）
1. はじめに
Dockerを使って、Stable Diffusion(ディープラーニングによるテキスト→画像作成ツール）を構築します。
本内容は、WindowsでGPUを使用して動かす方法について記載しています。
GPUを搭載していない場合でも、処理は遅くなりますがStable Diffusionを動かすことができます。
GPUを使用しない場合は以下を参照ください。
2. 事前準備
Windows版は以下の環境で実施しました。
| OS | Windows 11 Home |
| GPU | nVIDIA Geforce RTX3060 12GB |
| メモリ | 16GB |
2.1 Docker Desktopのダウンロード、インストール
以下URLからインストーラ(.exe)をダウンロードします。
ダウンロードしたファイルを実行し、Docker Desktopをインストールします。
Docker DesktopにはDocker Composeも含まれています。
2.2 Gitのダウンロード、インストール
以下URLからインストーラ(.exe)をダウンロードします。 ダウンロードしたファイルを実行し、Gitをインストールします。
2.3 NVIDIA CUDA Toolkitのインストール
以下URLからNVIDIA CUDA Toolkitをインストールします。
23/2/14現在 最新版はCUDA 12.0.0でしたが、
12.0.0では私の環境ではインストールできなかったので、CUDA 11.8.0（ひとつ前）をダウンロードしました。
以下から「CUDA Toolkit 11.8.0」のリンクをたどり、ダウンロードしました。
私のPC環境でインストール前のnVIDIAのドライバのバージョンが472.12だったのが関係しているのかもしれません。
(参考)NVIDIA CUDA Toolkitの最新版インストール。
2.4 wslの設定変更
StableDiffusionなど、メモリを大量に使用するコンテナを動かす場合は、メモリの割当を増やす必要があります。
LinuxではDocker Desktopの設定でメモリの変更ができましたが、Windows版ではwslのコンフィグファイルに直接記載する必要がありました。
「C:\Users\【ユーザ名】\.wslconfig」ファイルを新しく作成し、
[wsl2]
memory=13GB
swap=128GB
の記述を追加します。(メモリとスワップの値を大きく取ります。)
.wslconfig ファイルを変更したら一度PCを再起動します。
3. Stable Diffusion web UI (AUTOMATIC1111)のダウンロード
端末より以下のコマンドを実行し、Stable Diffusion WebUI Dockerをcloneします。
git clone https://github.com/AbdBarho/stable-diffusion-webui-docker
4. Stable Diffusionのコンテナを定義
端末より以下のコマンドを実行し、Stable Diffusion のコンテナを定義します。
cd stable-diffusion-webui-docker
docker compose --profile download up --build
正常に終了しない場合、2行目のコマンドを再度実行することで上手く行くことがあります。
5. Stable Diffusionを実行
端末より以下のコマンドを実行し、Stable Diffusion（GPU版） を実行します。
docker compose --profile auto up --build
Docker DesktopでStable Diffusionが動いていることを確認します。
6. ブラウザから起動
ブラウザより以下のアドレスを入力します。
http://localhost:7860
7. トラブルシューティング
Stable Diffusion を動かしているときにDockerDesktopでエラーコード137（以下の画面）が出る場合は、以下が参考になります。
7.1 GPUがDocker内で認識できているかの確認
端末から以下のコマンドを実行することで、nVIDIAがDocker内で認識できているか確認することができます。
docker run -it --gpus=all --rm nvidia/cuda:11.8.0-base-ubuntu22.04 /bin/bash
nvidia-smi
表示例ですが、以下のように表示されていればDocker内で認識できています。
7.2 GPUメモリのカスタマイズ
GPUのメモリが足りない場合、以下のホームページが参考になるかもしれません。
7.3 Dockerで使用するメモリのカスタマイズ
Dockerで使用するメモリを増やすことで上手く行ったことがあります。（上記「wslの設定変更」を参照）
.wslconfigの内容については以下ホームページも参考になりました。
更新履歴
2023/3/31 wslconfigファイルのサンプルの記載を削除（あまり重要ではないため）
2023/2/14 公開
Discussion