diff --git a/comfy3d-pt22/Dockerfile b/comfy3d-pt22/Dockerfile
index ca07203..17e01df 100644
--- 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
index 3221a57..a55fa36 100644
--- 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
index 1d14207..b9bd16d 100644
--- 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
index cb72a71..eb09c37 100644
--- 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