diff --git a/Dockerfile b/Dockerfile
index 9a2af0c..1872f91 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,7 +3,7 @@ WORKDIR /app
 RUN git clone --depth 1 --branch v0.3.67 https://github.com/comfyanonymous/ComfyUI.git ComfyUI && \
     rm -rf ComfyUI/.git
 
-FROM pytorch/pytorch:2.9.0-cuda13.0-cudnn9-runtime
+FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime
 WORKDIR /app
 COPY --from=builder /app/ComfyUI /app/ComfyUI
 WORKDIR /app/ComfyUI