diffusers/docker/diffusers-pytorch-cpu/Dockerfile at main · feisan/diffusers · GitHub

History

43 lines (38 loc) · 1.14 KB

Raw

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)

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

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

torch \

torchvision \

torchaudio \

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

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

accelerate \

datasets \

hf-doc-builder \

huggingface-hub \

Jinja2 \

librosa \

numpy \

scipy \

tensorboard \

transformers

CMD ["/bin/bash"]

diffusers/docker/diffusers-pytorch-cpu/Dockerfile at main · feisan/diffusers · GitHub

Expand file tree

Breadcrumbs

Dockerfile

Latest commit

History

Breadcrumbs

Dockerfile

File metadata and controls