From 1e602fe65e84d210bae10eb0fe38f6128254035d Mon Sep 17 00:00:00 2001 From: YAN Wenkun Date: Thu, 14 Nov 2024 05:51:07 +0800 Subject: [PATCH] 'comfy3d-pt25' further break down installation of PyTorch --- comfy3d-pt25/Dockerfile | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/comfy3d-pt25/Dockerfile b/comfy3d-pt25/Dockerfile index aa0ce16..33323f8 100644 --- 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 \