Stable DiffusionをDockerで動かすチュートリアル

環境

必要なものは以下の三つです。ちなみに環境は Ubuntu 22.04 です。

CUDA
Docker
Docker Compose

CUDA

1
$ nvidia-smi
2
+---------------------------------------------------------------------------------------+
3
| NVIDIA-SMI 530.41.03              Driver Version: 530.41.03    CUDA Version: 12.1     |
4
|-----------------------------------------+----------------------+----------------------+
5
| GPU  Name                  Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
6
| Fan  Temp  Perf            Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
7
|                                         |                      |               MIG M. |
8
|=========================================+======================+======================|
9
|   0  NVIDIA GeForce RTX 4070 Ti      Off| 00000000:01:00.0 Off |                  N/A |
10
|  0%   33C    P8                9W / 285W|   3043MiB / 12282MiB |      0%      Default |
11
|                                         |                      |                  N/A |
12
+-----------------------------------------+----------------------+----------------------+
13

14
+---------------------------------------------------------------------------------------+
15
| Processes:                                                                            |
16
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
17
|        ID   ID                                                             Usage      |
18
|=======================================================================================|
19
|    0   N/A  N/A      1156      G   /usr/lib/xorg/Xorg                          475MiB |
20
|    0   N/A  N/A      1376      G   /usr/bin/gnome-shell                         45MiB |
21
|    0   N/A  N/A      4887      C   python                                     2518MiB |
22
+---------------------------------------------------------------------------------------+

ドライバー自体は Ubuntu の Software Updater から導入しました。このままでは Docker のコンテナ内から CUDA が利用できないので、NVIDIA container toolkit を使って、docker のコンテナ上で cuda を動かすの記事を参考に NVIDIA container toolkit を導入してください。

Docker

1
$ docker -v
2
Docker version 23.0.6, build ef23cbc

Docker Compose

1
$ docker compose version
2
Docker Compose version v2.10.2

Stable Diffusion WebUI Docker

GitHub で公開されているのでこれを利用します。

1
git clone https://github.com/AbdBarho/stable-diffusion-webui-docker
2
cd stable-diffusion-webui-docker

ここに既にdocker-compose.ymlがあるのでこれを利用します。

1
docker compose --profile auto up --build

とすれば AUTOMATIC1111 版の Stable Diffusion WebUI が起動します。他にもいろいろ UI があるのですが、とりあえずこれでいいと思います。

するとビルドから何まですべてやってくれます。container toolkit を入れ忘れているとここでなんかエラーが出ます。

で、このままだとlocalhost:7860でしかアクセスできないので、外からアクセスできるようにします。ルーターのポートフォワードなどは各自やっているものとします。

docker-compose.yml

以下の内容をdocker-compose.ymlに追記します。いつも使っているnginx-proxyとletsencrypt-nginx-proxy-companionの組み合わせです。

TSL が必須でなければ後者は不要です。

またautoのところをちょっと変えます。portsが本当に必要かどうかはわからないので、有識者教えて下さい。

CLI_ARGSで実行時に渡されるオプションを変えられるので、xformersをとりあえず有効化しておきます（多分デフォルトで有効です）。

これを有効にすると速度がちょっと速くなる代わりに同一シードでの完全な再現性がなくなります。なのでお好みで。

1
  auto: &automatic
2
    <<: *base_service
3
    profiles: ["auto"]
4
    build: ./services/AUTOMATIC1111
5
    image: sd-auto:55
6
    ports:
7
      - '7860:7860'
8
    environment:
9
      VIRTUAL_HOST: $LETSENCRYPT_HOST
10
      VIRTUAL_POST: 7860
11
      LETSENCRYPT_HOST: $LETSENCRYPT_HOST
12
      LETSENCRYPT_EMAIL: $LETSENCRYPT_EMAIL
13
      CLI_ARGS: --allow-code --enable-insecure-extension-access --api --xformers --opt-sdp-attention --no-half-vae
14

15
  nginx-proxy:
16
    image: jwilder/nginx-proxy
17
    container_name: nginx-proxy
18
    profiles: ["external"]
19
    ports:
20
      - 80:80
21
      - 443:443
22
    volumes:
23
      - /var/run/docker.sock:/tmp/docker.sock:ro
24
      - ./certs:/etc/nginx/certs:ro
25
      - /etc/nginx/vhost.d
26
      - /usr/share/nginx/html
27
    restart: always
28
    logging:
29
      driver: none
30

31
  letsencrypt:
32
    image: jrcs/letsencrypt-nginx-proxy-companion
33
    container_name: letsencrypt
34
    profiles: ["external"]
35
    volumes:
36
      - /var/run/docker.sock:/var/run/docker.sock:ro
37
      - ./certs:/etc/nginx/certs:rw
38
    volumes_from:
39
      - nginx-proxy
40
    restart: always

環境変数から値を読み取っているので.envを作成してください。

1
LETSENCRYPT_HOST=
2
LETSENCRYPT_EMAIL=

二つだけ指定しておけば大丈夫です。

Makefile

自分は何度もコマンドを打つのがめんどくさかったので Makefile 化しました。

ローカルで動かしたいときならmake up、外部からも使えるようにしたいときはmake startという感じです。

1
.PHONY: up
2
up:
3
  docker compose --profile auto up --build
4

5
.PHONY: start
6
start:
7
  docker compose --profile auto --profile external up --build -d
8

9
.PHONY: down
10
down:
11
  docker compose down

VScode であれば SSH で繋げばそのへんもよしなにやってくれるので実は外部に公開しなくても SSH で繋げばローカル扱いでアクセスできます。

なのでこれは完全に第三者に公開するためのものです。

ここまで書いてあれですが、第三者に公開する必要がないならここまでの手順は全部要らないです。Docker で動かして終わりです。

記事は以上。