From 5fc3c539a2c297ee6c9d6e4e93b16578919b6ea9 Mon Sep 17 00:00:00 2001 From: YAN Wenkun Date: Thu, 14 Nov 2024 07:06:21 +0800 Subject: [PATCH] 'cu121's revert useless optimization --- comfy3d-pt22/Dockerfile | 10 +--------- comfy3d-pt23/Dockerfile | 10 +--------- cu121-megapak/Dockerfile | 24 ------------------------ cu121/Dockerfile | 24 ------------------------ 4 files changed, 2 insertions(+), 66 deletions(-) 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