mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-04-15 13:02:35 +08:00
97b4d164ed
Switch to a two-stage Dockerfile that builds SageAttention 2.2 from source on python:3.12-slim-trixie by explicitly enabling contrib/non-free/non-free-firmware in APT and installing Debian’s nvidia-cuda-toolkit (nvcc) for compilation, then installs the produced cp312 wheel into the slim runtime so --use-sage-attention works at startup. The builder installs Torch cu129 to match the runtime for ABI compatibility and uses pip’s --break-system-packages to avoid a venv while respecting PEP 668 in a controlled way, keeping layers lean and avoiding the prior sources.list and space issues seen on GitHub runners. The final image remains minimal while bundling an up-to-date SageAttention build aligned with the Torch/CUDA stack in use.
78 lines
3.0 KiB
Docker
78 lines
3.0 KiB
Docker
# --------------------------
|
|
# Stage 1: build SageAttention 2.2 wheel (Debian trixie + nvcc)
|
|
# --------------------------
|
|
FROM python:3.12.11-slim-trixie AS sage-builder
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive \
|
|
PIP_DISABLE_PIP_VERSION_CHECK=1 \
|
|
PIP_NO_CACHE_DIR=1
|
|
|
|
# Write explicit Debian sources with contrib/non-free/non-free-firmware, then install CUDA toolkit + build deps
|
|
RUN set -eux; \
|
|
printf 'deb http://deb.debian.org/debian trixie main contrib non-free non-free-firmware\n' > /etc/apt/sources.list; \
|
|
printf 'deb http://deb.debian.org/debian trixie-updates main contrib non-free non-free-firmware\n' >> /etc/apt/sources.list; \
|
|
printf 'deb http://security.debian.org/debian-security trixie-security main contrib non-free non-free-firmware\n' >> /etc/apt/sources.list; \
|
|
apt-get update; \
|
|
apt-get install -y --no-install-recommends ca-certificates curl git build-essential cmake nvidia-cuda-toolkit; \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
WORKDIR /tmp/sage
|
|
|
|
# Install Torch cu129 in builder (matches runtime) and build wheel (PEP 668: allow system installs here)
|
|
RUN python -m pip install --upgrade pip setuptools wheel --break-system-packages && \
|
|
python -m pip install torch torchvision torchaudio \
|
|
--extra-index-url https://download.pytorch.org/whl/cu129 \
|
|
--break-system-packages
|
|
|
|
# Shallow clone SageAttention and build a cp312 wheel
|
|
RUN git clone --depth 1 https://github.com/thu-ml/SageAttention.git . && \
|
|
python -m pip wheel . --no-deps --no-build-isolation -w /dist --break-system-packages
|
|
|
|
# --------------------------
|
|
# Stage 2: runtime image (slim)
|
|
# --------------------------
|
|
FROM python:3.12.11-slim-trixie
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive \
|
|
PYTHONUNBUFFERED=1 \
|
|
COMFY_AUTO_INSTALL=1 \
|
|
PIP_DISABLE_PIP_VERSION_CHECK=1 \
|
|
PIP_NO_CACHE_DIR=1
|
|
|
|
# System deps
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
git build-essential cmake libgl1 libglx-mesa0 libglib2.0-0 \
|
|
fonts-dejavu-core fontconfig util-linux \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Create runtime user/group
|
|
RUN groupadd --gid 1000 appuser \
|
|
&& useradd --uid 1000 --gid 1000 --create-home --shell /bin/bash appuser
|
|
|
|
WORKDIR /app/ComfyUI
|
|
|
|
# Install core deps (Torch cu129 must match builder)
|
|
COPY requirements.txt ./
|
|
RUN python -m pip install --upgrade pip setuptools wheel && \
|
|
python -m pip install torch torchvision torchaudio \
|
|
--extra-index-url https://download.pytorch.org/whl/cu129 && \
|
|
python -m pip install -r requirements.txt && \
|
|
python -m pip install imageio-ffmpeg "av>=14.2" nvidia-ml-py
|
|
|
|
# Install the SageAttention wheel built in the builder stage
|
|
COPY --from=sage-builder /dist/sageattention-*.whl /tmp/
|
|
RUN python -m pip install /tmp/sageattention-*.whl
|
|
|
|
# Copy the application
|
|
COPY . .
|
|
|
|
# Entrypoint and launch
|
|
COPY entrypoint.sh /entrypoint.sh
|
|
RUN chmod +x /entrypoint.sh \
|
|
&& chown -R appuser:appuser /app /home/appuser /entrypoint.sh
|
|
|
|
EXPOSE 8188
|
|
USER root
|
|
ENTRYPOINT ["/entrypoint.sh"]
|
|
CMD ["python", "main.py", "--listen", "0.0.0.0", "--use-sage-attention"]
|