mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-13 01:47:14 +08:00

The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface.

ai comfy comfyui python pytorch stable-diffusion

Go to file

clsferguson 1886bd4b96 build(docker): add CUDA 12.9 multi-stage; bake SageAttention 2.2 Switch from python:3.12-slim-trixie to a multi-stage NVIDIA CUDA 12.9 Ubuntu 22.04 build: use devel for compile (nvcc) and runtime for final image. Compile SageAttention 2.2+ from upstream source during image build by resolving the latest commit and installing without build isolation for a deterministic wheel. Install Triton (>=3.0.0) alongside Torch cu129 and start ComfyUI with --use-sage-attention by default. Add SAGE_FORCE_REFRESH build-arg to re-resolve the ref and bust cache when needed. This improves reproducibility, reduces startup latency, and keeps nvcc out of production for a smaller final image.		2025-09-22 06:30:25 -06:00
.ci
.github	Update build-release.yml	2025-09-21 22:46:37 -06:00
alembic_db
api_server
app
comfy	Hunyuan refiner vae now works with tiled. (#9836 )	2025-09-12 19:46:46 -04:00
comfy_api
comfy_api_nodes
comfy_config
comfy_execution
comfy_extras
custom_nodes
input
middleware
models
output
script_examples
tests
tests-unit
utils
.gitattributes
.gitignore
alembic.ini
CODEOWNERS
comfyui_version.py
CONTRIBUTING.md
cuda_malloc.py
Dockerfile	build(docker): add CUDA 12.9 multi-stage; bake SageAttention 2.2	2025-09-22 06:30:25 -06:00
entrypoint.sh	chore(bootstrap): trace root-only setup via run()	2025-09-17 14:49:01 -06:00
execution.py
extra_model_paths.yaml.example
folder_paths.py
hook_breaker_ac10a0.py
latent_preview.py
LICENSE
main.py
new_updater.py
node_helpers.py
nodes.py
protocol.py
pyproject.toml
pytest.ini
README.md	docs(readme): document baked-in SageAttention 2.2 and default enable via --use-sage-attention	2025-09-21 21:12:04 -06:00
requirements.txt
server.py

README.md

ComfyUI-Docker

An automated Repo for ComfyUI Docker image builds, optimized for NVIDIA GPUs.

About • Features • Getting Started • Usage • License

About

This repository automates the creation of Docker images for ComfyUI, a powerful and modular stable diffusion GUI and backend. It syncs with the upstream ComfyUI repository, builds a Docker image on new releases, and pushes it to GitHub Container Registry (GHCR).

I created this repo for myself as a simple way to stay up to date with the latest ComfyUI versions while having an easy-to-use Docker image. It's particularly suited for setups with NVIDIA GPUs, leveraging CUDA for accelerated performance.

Built With

Docker
GitHub Actions for automation
PyTorch with CUDA support
Based on Python 3.12 slim image

Features

Automated Sync & Build: Daily checks for upstream releases, auto-merges changes, and builds/pushes Docker images.
NVIDIA GPU Ready: Pre-configured with CUDA-enabled PyTorch for seamless GPU acceleration.
Non-Root Runtime: Runs as a non-root user for better security.
Pre-Installed Manager: Includes ComfyUI-Manager for easy node/extensions management.
SageAttention 2.2 baked in: The image compiles SageAttention 2.2/2++ from the upstream repository during docker build, ensuring the latest kernels for modern NVIDIA GPUs are included by default.
Auto-enabled at launch: ComfyUI is started with the --use-sage-attention flag so SageAttention is activated automatically on startup (no extra steps required).

Getting Started

Prerequisites

Docker: Installed on your host (e.g., Docker Desktop or Engine).
NVIDIA GPU: For GPU support (ensure NVIDIA drivers and CUDA are installed on the host).
NVIDIA Container Toolkit: For GPU passthrough in Docker (install via the official guide).

Pulling the Image

The latest image is available on GHCR:

docker pull ghcr.io/clsferguson/comfyui-docker:latest

For a specific version (synced with upstream tags, starting at 0.3.57):

docker pull ghcr.io/clsferguson/comfyui-docker:vX.Y.Z

Docker Compose

For easier management, use this docker-compose.yml:

services:
  comfyui:
    image: ghcr.io/clsferguson/comfyui-docker:latest
    container_name: ComfyUI
    runtime: nvidia
    restart: unless-stopped
    ports:
      - 8188:8188
    environment:
      - TZ=America/Edmonton
      - PUID=1000
      - GUID=1000
    gpus: all
    volumes:
      - comfyui_data:/app/ComfyUI/user/default
      - comfyui_nodes:/app/ComfyUI/custom_nodes
      - /mnt/comfyui/models:/app/ComfyUI/models
      - /mnt/comfyui/input:/app/ComfyUI/input
      - /mnt/comfyui/output:/app/ComfyUI/output

Run with docker compose up -d.

Usage

Basic Usage

Access ComfyUI at http://localhost:8188 after starting the container using Docker Compose.

SageAttention

SageAttention 2.2 is built into the image and enabled automatically on startup via --use-sage-attention.
To verify, check the container logs on startup; ComfyUI will print a line indicating SageAttention is active.

Environment Variables

Set via .env file or -e flags in docker compose or docker run.

License

Distributed under the MIT License (same as upstream ComfyUI). See LICENSE for more information.

Contact

Creator: clsferguson - GitHub
Project Link: https://github.com/clsferguson/ComfyUI-Docker

Built with ❤️ for easy AI workflows.