Skip to content

GitLab

Unverified Commit 869bad3e authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

FIx torch and cuda version in ONNX tests (#7164) 



update
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 01ee0978
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
docker/diffusers-onnxruntime-cuda/Dockerfile
View file @ 869bad3e
FROM nvidia/cuda:11.6.2-cudnn8-devel-ubuntu20.04 FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04
LABEL maintainer="Hugging Face" LABEL maintainer="Hugging Face"
LABEL repository="diffusers" LABEL repository="diffusers"
...@@ -24,9 +24,9 @@ ENV PATH="/opt/venv/bin:$PATH" ...@@ -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) # 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 && \ RUN python3 -m pip install --no-cache-dir --upgrade pip uv==0.1.11 && \
python3 -m uv pip install --no-cache-dir \ python3 -m uv pip install --no-cache-dir \
torch==2.1.2 \ torch \
torchvision==0.16.2 \ torchvision \
torchaudio==2.1.2 \ torchaudio \
"onnxruntime-gpu>=1.13.1" \ "onnxruntime-gpu>=1.13.1" \
--extra-index-url https://download.pytorch.org/whl/cu117 && \ --extra-index-url https://download.pytorch.org/whl/cu117 && \
python3 -m uv pip install --no-cache-dir \ python3 -m uv pip install --no-cache-dir \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment