From 2c669e84808f40ac5b5026fa44e948cfbc325119 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Wed, 28 Jan 2026 17:22:27 +0530 Subject: [PATCH] change to CUDA 12.9. (#13045) * change to CUDA 12.9. * up * change runtime base * FROM --- docker/diffusers-pytorch-cuda/Dockerfile | 12 ++++++++---- docker/diffusers-pytorch-xformers-cuda/Dockerfile | 12 ++++++++---- 2 files changed, 16 insertions(+), 8 deletions(-) diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index 134b47215d..0c639f8532 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -1,8 +1,8 @@ -FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04 +FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" -ARG PYTHON_VERSION=3.11 +ARG PYTHON_VERSION=3.10 ENV DEBIAN_FRONTEND=noninteractive RUN apt-get -y update \ @@ -36,8 +36,12 @@ ENV PATH="$VIRTUAL_ENV/bin:$PATH" RUN uv pip install --no-cache-dir \ torch \ torchvision \ - torchaudio \ - --index-url https://download.pytorch.org/whl/cu121 + torchaudio + +# Install compatible versions of numba/llvmlite for Python 3.10+ +RUN uv pip install --no-cache-dir \ + "llvmlite>=0.40.0" \ + "numba>=0.57.0" RUN uv pip install --no-cache-dir "git+https://github.com/huggingface/diffusers.git@main#egg=diffusers[test]" diff --git a/docker/diffusers-pytorch-xformers-cuda/Dockerfile b/docker/diffusers-pytorch-xformers-cuda/Dockerfile index 7714821c82..f40df34a4c 100644 --- a/docker/diffusers-pytorch-xformers-cuda/Dockerfile +++ b/docker/diffusers-pytorch-xformers-cuda/Dockerfile @@ -1,8 +1,8 @@ -FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04 +FROM nvidia/cuda:12.9.0-runtime-ubuntu20.04 LABEL maintainer="Hugging Face" LABEL repository="diffusers" -ARG PYTHON_VERSION=3.11 +ARG PYTHON_VERSION=3.10 ENV DEBIAN_FRONTEND=noninteractive RUN apt-get -y update \ @@ -36,8 +36,12 @@ ENV PATH="$VIRTUAL_ENV/bin:$PATH" RUN uv pip install --no-cache-dir \ torch \ torchvision \ - torchaudio \ - --index-url https://download.pytorch.org/whl/cu121 + torchaudio + +# Install compatible versions of numba/llvmlite for Python 3.10+ +RUN uv pip install --no-cache-dir \ + "llvmlite>=0.40.0" \ + "numba>=0.57.0" RUN uv pip install --no-cache-dir "git+https://github.com/huggingface/diffusers.git@main#egg=diffusers[test]"