Skip to content

GitLab

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

pin torch version (#5297)

parent 84b82a6c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
docker/diffusers-pytorch-cuda/Dockerfile
View file @ d8d8b2ae
...@@ -25,8 +25,8 @@ ENV PATH="/opt/venv/bin:$PATH" ...@@ -25,8 +25,8 @@ 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 && \ RUN python3 -m pip install --no-cache-dir --upgrade pip && \
python3 -m pip install --no-cache-dir \ python3 -m pip install --no-cache-dir \
torch \ torch==2.0.1 \
torchvision \ torchvision==0.15.2 \
torchaudio \ torchaudio \
invisible_watermark && \ invisible_watermark && \
python3 -m pip install --no-cache-dir \ python3 -m 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