Description

• ComfyUI's docker compose for WSL2 (GPU support).
• The feature is that folders under volumes are treated as symbolic links in docker, so you only need to place local files in the volumes folder.

Prerequisites

• Windows 11(64GB)
• NVidia video card (RTX3060 12GB)
• WSL2 (32GB and operation confirmed on Ubuntu 20.04)

*Not confirmed in other configurations than those listed above.

Installing

• Build Docker with the following command

docker compose build --build-arg UID="$(id -u)" --build-arg GID="$(id -g)" 

• Place checkpoint files (*.ckpt and *.safetensors) in /volumes/models/checkpoints.
• Put vae files (*.pt and *.safetensors) in /volumes/models/vae.
• Place LoRA, LoCoN and LoHA in /volumes/models/lolas.
• Output files are placed in /volumes/output/.

Usage

• After build, do the following.

docker compose up -d

• After docker compose up -d, access the following.

http://localhost:8188

• docker compose down to exit.

Version

• 2024/08/12 fix: Add Flux required, remove version, update README.

Acknowledgments

• ComfyUI
• nVidia 525 + Cuda 11.8 + Python 3.10 + pyTorch GPU Docker image
• stable-diffusion-webui
• docker-composeを使う上でホストとコンテナのユーザーIDとグループIDを揃える