Bump base image to PyTorch 2.12.1 on CUDA 13.0

2026-07-18 10:03:40 -04:00
parent 86d74c266c
commit d18fb9b716
1 changed files with 2 additions and 1 deletions
@@ -1,5 +1,6 @@
-FROM docker.io/pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime
+FROM docker.io/pytorch/pytorch:2.12.1-cuda13.0-cudnn9-runtime
 ARG COMFYUI_VERSION=v0.28.0
+ENV PIP_BREAK_SYSTEM_PACKAGES=1
 RUN apt-get update && apt-get install -y --no-install-recommends \
    build-essential git tini \
    && rm -rf /var/lib/apt/lists/*