前準備

dockerをインストール

NVIDIA Container Toolkitをインストール

動作場所の作成

Dockerで動作させるための場所を作成します。今回は、~/localai/comfyui というディレクトリを作成して、そこにファイルを置くことにします。

$ mkdir -p ~/localai/comfyui

Dockerファイル

Dockerファイルを作成します。今回は、~/localai/comfyui/Dockerfile というファイルを作成して、以下の内容を記述します。

FROM nvidia/cuda:13.4.2-runtime-ubuntu26.04

ENV DEBIAN_FRONTEND=noninteractive

RUN apt-get update && apt-get install -y \
    git \
    python3 \
    python3-pip \
    python3-venv \
    libgl1 \
    libglib2.0-0 \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /workspace

RUN git clone https://github.com/comfyanonymous/ComfyUI.git /workspace/ComfyUI

WORKDIR /workspace/ComfyUI

RUN python3 -m venv /opt/venv
ENV PATH="/opt/venv/bin:$PATH"
RUN pip install --no-cache-dir --upgrade pip
RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
RUN pip install --no-cache-dir -r requirements.txt

EXPOSE 8188

CMD ["python3", "main.py", "--listen", "0.0.0.0"]

Docker composeファイル

Docker composeファイルを作成します。今回は、~/localai/comfyui/docker-compose.yml というファイルを作成して、以下の内容を記述します。

services:
  comfyui:
    build: .
    container_name: comfyui
    runtime: nvidia
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
    ports:
      - "8188:8188"
    volumes:
      - ./models/checkpoints:/workspace/ComfyUI/models/checkpoints
      - ./models/loras:/workspace/ComfyUI/models/loras
      - ./models/vae:/workspace/ComfyUI/models/vae
      - ./output:/workspace/ComfyUI/output
      - ./custom_nodes:/workspace/ComfyUI/custom_nodes
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

起動

$ cd ~/localai/comfyui
$ sudo docker compose up -d --build

動作確認

ブラウザで http://localhost:8188 にアクセスして、ComfyUIが表示されることを確認します。