'comfy3d-pt25' further break down installation of PyTorch

YanWenKun · Nov 13, 2024 · 1e602fe · 1e602fe
1 parent cebca45
commit 1e602fe
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/comfy3d-pt25/Dockerfile b/comfy3d-pt25/Dockerfile
@@ -162,6 +162,31 @@ RUN --mount=type=cache,target=/root/.cache/pip \
         --no-deps torch==2.5.1 \
         --index-url https://download.pytorch.org/whl/cu124
 
+RUN --mount=type=cache,target=/root/.cache/pip \
+    pip install \
+        --no-deps torch==2.5.1 nvidia-cudnn-cu12 \
+        --index-url https://download.pytorch.org/whl/cu124
+
+RUN --mount=type=cache,target=/root/.cache/pip \
+    pip install \
+        --no-deps torch==2.5.1 nvidia-cublas-cu12 \
+        --index-url https://download.pytorch.org/whl/cu124
+
+RUN --mount=type=cache,target=/root/.cache/pip \
+    pip install \
+        --no-deps torch==2.5.1 triton \
+        --index-url https://download.pytorch.org/whl/cu124
+
+RUN --mount=type=cache,target=/root/.cache/pip \
+    pip install \
+        --no-deps torch==2.5.1 nvidia-cusparse-cu12 \
+        --index-url https://download.pytorch.org/whl/cu124
+
+RUN --mount=type=cache,target=/root/.cache/pip \
+    pip install \
+        --no-deps torch==2.5.1 nvidia-nccl-cu12 \
+        --index-url https://download.pytorch.org/whl/cu124
+
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \
         xformers==0.0.28.post3 torch==2.5.1 torchvision torchaudio \