'cu121's revert useless optimization

YanWenKun · Nov 13, 2024 · 5fc3c53 · 5fc3c53
1 parent 32b2a51
commit 5fc3c53
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 66 deletions.
diff --git a/comfy3d-pt22/Dockerfile b/comfy3d-pt22/Dockerfile
@@ -169,15 +169,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \
-        --no-deps xformers==0.0.25.post1 torch==2.2.2 torchvision torchaudio
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps torch==2.2.2 nvidia-cudnn-cu12
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps torch==2.2.2 nvidia-cublas-cu12
+        --no-deps xformers==0.0.25.post1 torch==2.2.2
 
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \

diff --git a/comfy3d-pt23/Dockerfile b/comfy3d-pt23/Dockerfile
@@ -168,15 +168,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \
-        --no-deps xformers==0.0.26.post1 torch==2.3.0 torchvision torchaudio
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps torch==2.3.0 nvidia-cudnn-cu12
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps torch==2.3.0 nvidia-cublas-cu12
+        --no-deps xformers==0.0.26.post1 torch==2.3.0
 
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \

diff --git a/cu121-megapak/Dockerfile b/cu121-megapak/Dockerfile
@@ -163,34 +163,10 @@ RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \
         --dry-run xformers torchvision torchaudio
 
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps xformers torchvision torchaudio
-
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \
         --no-deps xformers torch torchvision torchaudio
 
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps xformers torch nvidia-cudnn-cu12
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps xformers torch nvidia-cublas-cu12
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps xformers torch triton
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps xformers torch nvidia-cusparse-cu12
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps xformers torch nvidia-nccl-cu12
-
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \
         xformers torchvision torchaudio

diff --git a/cu121/Dockerfile b/cu121/Dockerfile
@@ -43,34 +43,10 @@ RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \
         --dry-run xformers torchvision torchaudio
 
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps xformers torchvision torchaudio
-
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \
         --no-deps xformers torch torchvision torchaudio
 
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps xformers torch nvidia-cudnn-cu12
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps xformers torch nvidia-cublas-cu12
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps xformers torch triton
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps xformers torch nvidia-cusparse-cu12
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
-        --no-deps xformers torch nvidia-nccl-cu12
-
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \
         xformers torchvision torchaudio