mirror of
https://github.com/huggingface/diffusers.git
synced 2025-12-06 12:34:13 +08:00
57636ad4f4
* purge HF_HUB_ENABLE_HF_TRANSFER; promote Xet * purge HF_HUB_ENABLE_HF_TRANSFER; promote Xet x2 * restrict docker build test to the ones we actually use in CI. --------- Co-authored-by: YiYi Xu <yixu310@gmail.com> Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
49 lines
1.4 KiB
Docker
49 lines
1.4 KiB
Docker
FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04
|
|
LABEL maintainer="Hugging Face"
|
|
LABEL repository="diffusers"
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
RUN apt-get -y update \
|
|
&& apt-get install -y software-properties-common \
|
|
&& add-apt-repository ppa:deadsnakes/ppa
|
|
|
|
RUN apt install -y bash \
|
|
build-essential \
|
|
git \
|
|
git-lfs \
|
|
curl \
|
|
ca-certificates \
|
|
libsndfile1-dev \
|
|
libgl1 \
|
|
python3.10 \
|
|
python3-pip \
|
|
python3.10-venv && \
|
|
rm -rf /var/lib/apt/lists
|
|
|
|
# make sure to use venv
|
|
RUN python3.10 -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)
|
|
RUN python3.10 -m pip install --no-cache-dir --upgrade pip uv==0.1.11 && \
|
|
python3.10 -m uv pip install --no-cache-dir \
|
|
torch \
|
|
torchvision \
|
|
torchaudio \
|
|
"onnxruntime-gpu>=1.13.1" \
|
|
--extra-index-url https://download.pytorch.org/whl/cu117 && \
|
|
python3.10 -m uv pip install --no-cache-dir \
|
|
accelerate \
|
|
datasets \
|
|
hf-doc-builder \
|
|
huggingface-hub \
|
|
hf_xet \
|
|
Jinja2 \
|
|
librosa \
|
|
numpy==1.26.4 \
|
|
scipy \
|
|
tensorboard \
|
|
transformers
|
|
|
|
CMD ["/bin/bash"] |