46 lines
1.4 KiB
Docker
46 lines
1.4 KiB
Docker
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
|
|
|
|
# make sure to use venv
|
|
RUN python3 -m venv /opt/venv
|
|
ENV PATH="/opt/venv/bin:$PATH"
|
|
|
|
# pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)
|
|
# follow the instructions here: https://cloud.google.com/tpu/docs/run-in-container#train_a_jax_model_in_a_docker_container
|
|
RUN python3 -m pip install --no-cache-dir --upgrade pip && \
|
|
python3 -m pip install --no-cache-dir \
|
|
"jax[tpu]>=0.2.16,!=0.3.2" \
|
|
-f https://storage.googleapis.com/jax-releases/libtpu_releases.html && \
|
|
python3 -m pip install --upgrade --no-cache-dir \
|
|
clu \
|
|
"flax>=0.4.1" \
|
|
"jaxlib>=0.1.65" && \
|
|
python3 -m pip install --no-cache-dir \
|
|
accelerate \
|
|
datasets \
|
|
hf-doc-builder \
|
|
huggingface-hub \
|
|
Jinja2 \
|
|
librosa \
|
|
numpy \
|
|
scipy \
|
|
tensorboard \
|
|
transformers
|
|
|
|
CMD ["/bin/bash"] |