AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

change to CUDA 12.9. (#13045)

Browse Source 
* change to CUDA 12.9.

* up

* change runtime base

* FROM
This commit is contained in:
Sayak Paul
2026-01-28 17:22:27 +05:30
committed by GitHub
parent 2ac39ba664
commit 2c669e8480
2 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
docker/diffusers-pytorch-cuda/Dockerfile
View File

@@ -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]"

12
docker/diffusers-pytorch-xformers-cuda/Dockerfile
View File

@@ -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]"