Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 12 additions & 1 deletion services/comfy/complete/dockerfile.comfy.cuda.complete
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,20 @@ COPY --chown=comfy:comfy ./extra-requirements.txt ./

# Install additional python requirements
# Using the same cache path as the core image
#
# --extra-index-url cu130 is REQUIRED. The base image's PyTorch is built
# against CUDA 13.0 (torch==2.12.0+cu130). Without the cu130 index, pip
# only sees the stock cu128 torchaudio wheel on PyPI and will replace
# the cu130 one whenever a transitive dep triggers re-resolution,
# producing at runtime:
# RuntimeError: Detected that PyTorch and TorchAudio were compiled
# with different CUDA versions. PyTorch has CUDA version 13.0 whereas
# TorchAudio has CUDA version 12.8.
RUN --mount=type=cache,mode=0755,uid=1000,gid=1000,target=/app/.cache/pip \
source $VENV_PATH/bin/activate && \
pip install -r extra-requirements.txt
pip install \
--extra-index-url https://download.pytorch.org/whl/cu130 \
-r extra-requirements.txt

# Re-apply world-writable permissions after installing extra packages.
# The core stage's chmod doesn't cover files added in this layer — newly
Expand Down
Loading