Skip to content

YanWenKun/ComfyUI-Docker

Repository files navigation

Docker images for ComfyUI

English | 🀄 中文说明

This repo is for Docker images that runs ComfyUI - a Stable Diffusion GUI powering node-based workflow.

Quick Start - NVIDIA GPU

mkdir -p \
  storage \
  storage-models/models \
  storage-models/hf-hub \
  storage-models/torch-hub \
  storage-user/input \
  storage-user/output \
  storage-user/workflows

docker run -it --rm \
  --name comfyui \
  --runtime nvidia \
  --gpus all \
  -p 8188:8188 \
  -v "$(pwd)"/storage:/root \
  -v "$(pwd)"/storage-models/models:/root/ComfyUI/models \
  -v "$(pwd)"/storage-models/hf-hub:/root/.cache/huggingface/hub \
  -v "$(pwd)"/storage-models/torch-hub:/root/.cache/torch/hub \
  -v "$(pwd)"/storage-user/input:/root/ComfyUI/input \
  -v "$(pwd)"/storage-user/output:/root/ComfyUI/output \
  -v "$(pwd)"/storage-user/workflows:/root/ComfyUI/user/default/workflows \
  -e CLI_ARGS="--disable-xformers" \
  yanwk/comfyui-boot:cu128-slim

CUDA Compatibility

The supported CUDA versions for each GPU architecture are shown in the table below:

GPU Architecture Blackwell Hopper Ada Lovelace Ampere Turing Volta Pascal Maxwell

Example GPU

RTX 5090

H100

RTX 4090

RTX 3090

RTX 2080
GTX 1660

TITAN V

GTX 1080

GTX 980

cu130

✔️

✔️

✔️

✔️

✔️

cu128 ⭐

✔️

✔️

✔️

✔️

✔️

✔️

cu126

✔️

✔️

✔️

✔️

✔️

✔️

✔️

  • CUDA 12.8 images are currently recommended, in line with PyTorch’s development plan.

  • If you are unsure about your NVIDIA GPU architecture, see this article.

Note
These CUDA compatibility limitations are due to the PyTorch toolchain, not the NVIDIA CUDA Toolkit. For more information, refer to the PyTorch build script.

CUDA Image Tags

Slim

The slim images start with only ComfyUI and ComfyUI-Manager, yet include many dependencies to make future Custom Node installation easier. Recommended for beginners.

MEGAPAK

The megapak images are all-in-one bundles, including development kits and dozens of Custom Nodes for ComfyUI.

Testing

  • nightly

    • Using development version of PyTorch. For testing latest features.

More Image Tags

  • rocm

    • For AMD GPUs with ROCm 7 (latest version).

  • rocm6

    • For AMD GPUs with ROCm 6 (stable version).

  • xpu

    • For Intel GPUs with XPU.

  • cpu

    • A smaller image for CPU only.

  • archived

    • Archived Dockerfiles of retired image tags.

License

This open source license is written and valid both in Chinese and English, how good is that!

About

🐳Dockerfile for 🎨ComfyUI. | 容器镜像与启动脚本

Topics

Resources

License

Stars

Watchers

Forks

Sponsor this project