Skip to content

GitLab

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

Update base image for slow CUDA tests (#5426) 

update base image for tests
parent 324d18fb
Hide whitespace changes
Inline Side-by-side
Showing with 53 additions and 53 deletions
docker/diffusers-pytorch-compile-cuda/Dockerfile
View file @ 98a0712d
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"
...@@ -6,17 +6,17 @@ ENV DEBIAN_FRONTEND=noninteractive ...@@ -6,17 +6,17 @@ ENV DEBIAN_FRONTEND=noninteractive
RUN apt update && \ RUN apt update && \
apt install -y bash \ apt install -y bash \
build-essential \ build-essential \
git \ git \
git-lfs \ git-lfs \
curl \ curl \
ca-certificates \ ca-certificates \
libsndfile1-dev \ libsndfile1-dev \
libgl1 \ libgl1 \
python3.9 \ python3.9 \
python3.9-dev \ python3.9-dev \
python3-pip \ python3-pip \
python3.9-venv && \ python3.9-venv && \
rm -rf /var/lib/apt/lists rm -rf /var/lib/apt/lists
# make sure to use venv # make sure to use venv
...@@ -26,21 +26,21 @@ ENV PATH="/opt/venv/bin:$PATH" ...@@ -26,21 +26,21 @@ 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.9 -m pip install --no-cache-dir --upgrade pip && \ RUN python3.9 -m pip install --no-cache-dir --upgrade pip && \
python3.9 -m pip install --no-cache-dir \ python3.9 -m pip install --no-cache-dir \
torch \ torch \
torchvision \ torchvision \
torchaudio \ torchaudio \
invisible_watermark && \ invisible_watermark && \
python3.9 -m pip install --no-cache-dir \ python3.9 -m pip install --no-cache-dir \
accelerate \ accelerate \
datasets \ datasets \
hf-doc-builder \ hf-doc-builder \
huggingface-hub \ huggingface-hub \
Jinja2 \ Jinja2 \
librosa \ librosa \
numpy \ numpy \
scipy \ scipy \
tensorboard \ tensorboard \
transformers \ transformers \
omegaconf omegaconf
CMD ["/bin/bash"] CMD ["/bin/bash"]
docker/diffusers-pytorch-cuda/Dockerfile
View file @ 98a0712d
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"
...@@ -6,16 +6,16 @@ ENV DEBIAN_FRONTEND=noninteractive ...@@ -6,16 +6,16 @@ ENV DEBIAN_FRONTEND=noninteractive
RUN apt update && \ RUN apt update && \
apt install -y bash \ apt install -y bash \
build-essential \ build-essential \
git \ git \
git-lfs \ git-lfs \
curl \ curl \
ca-certificates \ ca-certificates \
libsndfile1-dev \ libsndfile1-dev \
libgl1 \ libgl1 \
python3.8 \ python3.8 \
python3-pip \ python3-pip \
python3.8-venv && \ python3.8-venv && \
rm -rf /var/lib/apt/lists rm -rf /var/lib/apt/lists
# make sure to use venv # make sure to use venv
...@@ -25,21 +25,21 @@ ENV PATH="/opt/venv/bin:$PATH" ...@@ -25,21 +25,21 @@ 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 \
torchvision \ torchvision \
torchaudio \ torchaudio \
invisible_watermark && \ invisible_watermark && \
python3 -m pip install --no-cache-dir \ python3 -m pip install --no-cache-dir \
accelerate \ accelerate \
datasets \ datasets \
hf-doc-builder \ hf-doc-builder \
huggingface-hub \ huggingface-hub \
Jinja2 \ Jinja2 \
librosa \ librosa \
numpy \ numpy \
scipy \ scipy \
tensorboard \ tensorboard \
transformers \ transformers \
omegaconf omegaconf
CMD ["/bin/bash"] CMD ["/bin/bash"]
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