FROM ubuntu:20.04

LABEL maintainer="Hugging Face"

LABEL repository="diffusers"

ENV DEBIAN_FRONTEND=noninteractive

RUN apt update && \

apt install -y bash \

build-essential \

git \

git-lfs \

curl \

ca-certificates \

libsndfile1-dev \

python3.8 \

python3-pip \

python3.8-venv && \

rm -rf /var/lib/apt/lists

RUN python3 -m venv /opt/venv

ENV PATH="/opt/venv/bin:$PATH"

RUN python3 -m pip install --no-cache-dir --upgrade pip && \

python3 -m pip install --no-cache-dir \

torch \

torchvision \

torchaudio \

onnxruntime \

--extra-index-url https://download.pytorch.org/whl/cpu && \

python3 -m pip install --no-cache-dir \

accelerate \

datasets \

hf-doc-builder \

huggingface-hub \

librosa \

modelcards \

numpy \

scipy \

tensorboard \

transformers

CMD ["/bin/bash"]