AI-Dock + ComfyUI Docker 镜像

在高度可配置、以云为中心的 AI-Dock 容器中运行 ComfyUI。

[!注意] 这些镜像不包含模型或第三方配置。您应该使用一个 配置脚本 来自动配置您的容器。您可以在 config/provisioning 中找到示例，包括 SD3 和 FLUX.1 的设置。

文档

所有 AI-Dock 容器都共享一个通用的基础镜像，旨在使在诸如 vast.ai 等云服务上运行尽可能简单且用户友好。

常见功能和选项已在 基础 Wiki 中记录，而此镜像特有的任何附加功能将在下文详细说明。

版本标签

:latest 标签指向 :latest-cuda，并对应于稳定且经过测试的版本。可能存在更新的构建版本。

标签遵循以下模式：

CUDA

• :cuda-[x.x.x-base|runtime]-[ubuntu-version]

ROCm

• :rocm-[x.x.x-runtime]-[ubuntu-version]

CPU

• :cpu-[ubuntu-version]

请浏览 ghcr.io 以查找适合您目标环境的镜像。或者，您也可以在 DockerHub 上查看精选的 CUDA 和 ROCm 构建版本。

支持平台：NVIDIA CUDA、AMD ROCm、CPU

附加环境变量

变量	描述
AUTO_UPDATE	启动时更新 ComfyUI（默认为 false）
CIVITAI_TOKEN	对来自 Civitai 的下载请求进行身份验证——对于受限制的模型是必需的
COMFYUI_ARGS	启动参数。例如：--gpu-only --highvram
COMFYUI_PORT_HOST	ComfyUI 界面端口（默认为 8188）
COMFYUI_REF	自动更新的 Git 引用。接受分支、标签或提交哈希值。默认：最新发布版本
COMFYUI_URL	使用 URL 覆盖 $DIRECT_ADDRESS:port 以指向 ComfyUI
HF_TOKEN	对来自 HuggingFace 的下载请求进行身份验证——对于受限制的模型（SD3、FLUX 等）是必需的

有关更多配置选项，请参阅基础环境变量 此处。

附加 Python 环境

环境	包括的软件包
comfyui	ComfyUI 及其依赖项
api	ComfyUI API 封装及其依赖项

登录 Shell 时将激活 comfyui 环境。

请参阅基础 micromamba 环境 此处。

附加服务

以下服务将与基础镜像提供的 默认服务 一同启动。

ComfyUI

该服务将在端口 8188 上启动，除非您已使用 COMFYUI_PORT_HOST 指定了覆盖端口。

您可以通过变量 COMFYUI_ARGS 设置启动标志。

要管理此服务，您可以使用 supervisorctl [start|stop|restart] comfyui。

ComfyUI API 封装

该服务在端口 8188 上可用，目前仍在开发中，旨在取代已被弃用的旧有的无服务器处理程序；如果您需要，旧版 Docker 镜像和源代码仍然可用。

您可以直接通过 /ai-dock/api/ 访问 API，也可以使用位于 /ai-dock/api/docs 的 Swager/openAPI 演示页面。

[!注意] 所有服务默认受密码保护。有关更多信息，请参阅 安全性 和 环境变量 文档。

预配置模板

Vast.ai

• comfyui:latest-cuda

• comfyui:latest-cuda + FLUX.1

• comfyui:latest-rocm

---

作者 (@robballantyne) 在您注册本文档中链接的服务时可能会获得报酬。在多种不同环境中测试多个 GPU 镜像变体既昂贵又耗时；这笔费用有助于抵消相关成本。

ComfyUI (AI-Dock) 快速上手指南

本指南基于 ai-dock/comfyui Docker 镜像，帮助开发者在云端或本地快速部署高度可配置的 ComfyUI 环境。该镜像支持 NVIDIA CUDA、AMD ROCm 及 CPU 模式，专为云服务商（如 vast.ai）优化。

1. 环境准备

系统要求

• 操作系统: Linux (推荐 Ubuntu 20.04/22.04)，Windows 或 macOS 需安装 Docker Desktop。
• 硬件:
  • GPU 加速: 需要安装对应的显卡驱动。
    • NVIDIA: 需安装 NVIDIA Container Toolkit。
    • AMD: 需安装 ROCm 驱动及容器支持。
  • CPU 模式: 无特殊硬件要求，但推理速度较慢。
• 软件依赖:
  • Docker Engine
  • Docker Compose (可选，用于编排)

前置说明

• 该镜像不包含任何预训练的模型文件或第三方配置。
• 首次运行时，强烈建议通过配置脚本 (Provisioning Script) 自动下载模型（如 SD3, FLUX.1）和安装插件。
• 如需下载受限模型（Gated Models），请准备好 HuggingFace (HF_TOKEN) 或 Civitai (CIVITAI_TOKEN) 的访问令牌。

2. 安装步骤

拉取镜像

根据硬件环境选择合适的标签。latest 默认指向稳定的 CUDA 版本。

# NVIDIA GPU (推荐)
docker pull ghcr.io/ai-dock/comfyui:latest-cuda

# AMD ROCm
docker pull ghcr.io/ai-dock/comfyui:latest-rocm

# CPU 模式
docker pull ghcr.io/ai-dock/comfyui:latest-cpu

提示: 如果访问 GitHub Container Registry (ghcr.io) 较慢，可尝试从 DockerHub 拉取： docker pull aidockorg/comfyui-cuda:latest

启动容器

基础启动命令 (NVIDIA 示例)

以下命令将启动容器并映射 8188 端口：

docker run -d \
  --name comfyui \
  --gpus all \
  -p 8188:8188 \
  -e AUTO_UPDATE=true \
  -e HF_TOKEN=YOUR_HUGGINGFACE_TOKEN \
  -v $(pwd)/data:/workspace \
  ghcr.io/ai-dock/comfyui:latest-cuda

关键参数说明

• --gpus all: 启用 GPU 加速（ROCm 用户可能需要 --device /dev/kfd --device /dev/dri）。
• -p 8188:8188: 映射 Web 界面端口，可通过 COMFYUI_PORT_HOST 环境变量修改宿主机端口。
• -e AUTO_UPDATE=true: 容器启动时自动更新 ComfyUI 到最新版本。
• -e HF_TOKEN=...: 填入你的 HuggingFace Token，用于下载 SD3、FLUX 等受限模型。
• -v $(pwd)/data:/workspace: 持久化存储工作区数据（模型、输出图片、工作流）。

高级配置 (自定义启动参数)

如需指定显存策略或仅使用 GPU，可添加 COMFYUI_ARGS：

-e COMFYUI_ARGS="--gpu-only --highvram"

3. 基本使用

访问 Web 界面

容器启动后，在浏览器中访问：

http://localhost:8188

注意: 默认情况下服务受密码保护。请查阅 AI-Dock 基础文档中的 "Security" 部分设置 PASSWORD 环境变量，或在日志中查看初始密码。

管理服务

容器内使用 supervisorctl 管理 ComfyUI 服务：

# 进入容器
docker exec -it comfyui bash

# 重启 ComfyUI 服务
supervisorctl restart comfyui

# 查看服务状态
supervisorctl status comfyui

自动化配置模型 (推荐)

由于镜像不含模型，建议在挂载卷中创建 provision.sh 脚本并在启动时挂载，或使用 CIVITAI_TOKEN 配合官方提供的配置示例自动下载。

示例：在首次启动日志中观察模型下载进度，或通过 API 上传工作流。

使用 API

ComfyUI API 封装服务运行在同一端口，可用于无服务器调用：

• API 端点: http://localhost:8188/ai-dock/api/
• Swagger 文档: http://localhost:8188/ai-dock/api/docs

---

注：本镜像由 AI-Dock 维护，旨在简化云端部署流程。生产环境请务必配置好安全认证及数据持久化卷。