Merge branch 'main' into onnx-cuda-test-update

2025-12-08 05:24:20 +08:00 · 2024-03-04 11:32:18 +05:30 · 2024-03-04 08:33:32 +05:30 · 2024-03-01 13:47:01 +05:30 · 2024-03-01 07:19:25 +00:00
1 changed files with 4 additions and 4 deletions
--- a/docker/diffusers-onnxruntime-cuda/Dockerfile
+++ b/docker/diffusers-onnxruntime-cuda/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.6.2-cudnn8-devel-ubuntu20.04
+FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04
 LABEL maintainer="Hugging Face"
 LABEL repository="diffusers"

@@ -24,9 +24,9 @@ ENV PATH="/opt/venv/bin:$PATH"
 # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)
 RUN python3 -m pip install --no-cache-dir --upgrade pip uv==0.1.11 && \
    python3 -m uv pip install --no-cache-dir \
-        torch==2.1.2 \
-        torchvision==0.16.2 \
-        torchaudio==2.1.2 \
+        torch \
+        torchvision \
+        torchaudio \
        "onnxruntime-gpu>=1.13.1" \
        --extra-index-url https://download.pytorch.org/whl/cu117 && \
    python3 -m uv pip install --no-cache-dir \
Author	SHA1	Message	Date
Dhruv Nair	c68c139090	Merge branch 'main' into onnx-cuda-test-update	2024-03-04 11:32:18 +05:30
Sayak Paul	4a58ebece1	Merge branch 'main' into onnx-cuda-test-update	2024-03-04 08:33:32 +05:30
Sayak Paul	7511e98f18	Merge branch 'main' into onnx-cuda-test-update	2024-03-01 13:47:01 +05:30
Dhruv Nair	6379a91584	update	2024-03-01 07:19:25 +00:00