Skip to content

GitLab

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

Update docker image for xformers (#5597) 

update docker image for xformers
parent ce9484b1
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
docker/diffusers-pytorch-xformers-cuda/Dockerfile
View file @ f1d052c5
FROM nvidia/cuda:11.7.1-cudnn8-runtime-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"
...@@ -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==2.0.1 \ torch \
torchvision==0.15.2 \ torchvision \
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