From fb10f4889e90dea240530929d6b5be93fee01743 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 28 Jan 2026 18:23:47 +0530 Subject: [PATCH 1/4] uniform run times and wheels for pytorch cuda. --- docker/diffusers-pytorch-cuda/Dockerfile | 5 +++-- docker/diffusers-pytorch-xformers-cuda/Dockerfile | 5 +++-- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index 0c639f853248..320a133c325f 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 +FROM nvidia/cuda:12.8.0-runtime-ubuntu20.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" @@ -36,7 +36,8 @@ ENV PATH="$VIRTUAL_ENV/bin:$PATH" RUN uv pip install --no-cache-dir \ torch \ torchvision \ - torchaudio + torchaudio \ + --index-url https://download.pytorch.org/whl/cu128 # Install compatible versions of numba/llvmlite for Python 3.10+ RUN uv pip install --no-cache-dir \ diff --git a/docker/diffusers-pytorch-xformers-cuda/Dockerfile b/docker/diffusers-pytorch-xformers-cuda/Dockerfile index f40df34a4c59..9f9d4e29fc4b 100644 --- a/docker/diffusers-pytorch-xformers-cuda/Dockerfile +++ b/docker/diffusers-pytorch-xformers-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 +FROM nvidia/cuda:12.8.0-runtime-ubuntu20.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" @@ -36,7 +36,8 @@ ENV PATH="$VIRTUAL_ENV/bin:$PATH" RUN uv pip install --no-cache-dir \ torch \ torchvision \ - torchaudio + torchaudio \ + --index-url https://download.pytorch.org/whl/cu128 # Install compatible versions of numba/llvmlite for Python 3.10+ RUN uv pip install --no-cache-dir \ From 71bf49250bb5acb4a6cc4941194eea579cd2a859 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 28 Jan 2026 20:51:29 +0530 Subject: [PATCH 2/4] 12.9 --- docker/diffusers-pytorch-cuda/Dockerfile | 4 ++-- docker/diffusers-pytorch-xformers-cuda/Dockerfile | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index 320a133c325f..f507aa381b1b 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.8.0-runtime-ubuntu20.04 +FROM nvidia/cuda:12.9.1-runtime-ubuntu22.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" @@ -37,7 +37,7 @@ RUN uv pip install --no-cache-dir \ torch \ torchvision \ torchaudio \ - --index-url https://download.pytorch.org/whl/cu128 + --index-url https://download.pytorch.org/whl/cu129 # Install compatible versions of numba/llvmlite for Python 3.10+ RUN uv pip install --no-cache-dir \ diff --git a/docker/diffusers-pytorch-xformers-cuda/Dockerfile b/docker/diffusers-pytorch-xformers-cuda/Dockerfile index 9f9d4e29fc4b..e856efbe9283 100644 --- a/docker/diffusers-pytorch-xformers-cuda/Dockerfile +++ b/docker/diffusers-pytorch-xformers-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.8.0-runtime-ubuntu20.04 +FROM nvidia/cuda:12.9.1-runtime-ubuntu22.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" @@ -37,7 +37,7 @@ RUN uv pip install --no-cache-dir \ torch \ torchvision \ torchaudio \ - --index-url https://download.pytorch.org/whl/cu128 + --index-url https://download.pytorch.org/whl/cu129 # Install compatible versions of numba/llvmlite for Python 3.10+ RUN uv pip install --no-cache-dir \ From 21fe9aca93d0b9d14994253ec5519bdb6a1ae6f7 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Thu, 29 Jan 2026 13:30:14 +0530 Subject: [PATCH 3/4] change to 24.04. --- docker/diffusers-pytorch-cuda/Dockerfile | 2 +- docker/diffusers-pytorch-xformers-cuda/Dockerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index f507aa381b1b..22b18e2e8072 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.9.1-runtime-ubuntu22.04 +FROM FROM nvidia/cuda:12.9.1-runtime-ubuntu24.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" diff --git a/docker/diffusers-pytorch-xformers-cuda/Dockerfile b/docker/diffusers-pytorch-xformers-cuda/Dockerfile index e856efbe9283..b3ba11290d14 100644 --- a/docker/diffusers-pytorch-xformers-cuda/Dockerfile +++ b/docker/diffusers-pytorch-xformers-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.9.1-runtime-ubuntu22.04 +FROM FROM nvidia/cuda:12.9.1-runtime-ubuntu24.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" From 4306c113d1ece5514008e4579620e655c66d2851 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Thu, 29 Jan 2026 13:33:52 +0530 Subject: [PATCH 4/4] change to 24.04. --- docker/diffusers-pytorch-cuda/Dockerfile | 2 +- docker/diffusers-pytorch-xformers-cuda/Dockerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index 22b18e2e8072..b11d8a491168 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM FROM nvidia/cuda:12.9.1-runtime-ubuntu24.04 +FROM nvidia/cuda:12.9.1-runtime-ubuntu24.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" diff --git a/docker/diffusers-pytorch-xformers-cuda/Dockerfile b/docker/diffusers-pytorch-xformers-cuda/Dockerfile index b3ba11290d14..33cae319b93c 100644 --- a/docker/diffusers-pytorch-xformers-cuda/Dockerfile +++ b/docker/diffusers-pytorch-xformers-cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM FROM nvidia/cuda:12.9.1-runtime-ubuntu24.04 +FROM nvidia/cuda:12.9.1-runtime-ubuntu24.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers"