mirror of
https://github.com/huggingface/diffusers.git
synced 2025-12-16 01:14:47 +08:00
4e59bcc680
* [WIP][CI] Framework and hardware-specific docker images for CI tests * username * fix cpu * try out the image * push latest * update workspace * no root isolation for actions * add a flax image * flax and onnx matrix * fix runners * add reports * onnxruntime image * retry tpu * fix * fix * build onnxruntime * naming * onnxruntime-gpu image * onnxruntime-gpu image, slow tests * latest jax version * trigger flax * run flax tests in one thread * fast flax tests on cpu * fast flax tests on cpu * trigger slow tests * rebuild torch cuda * force cuda provider * fix onnxruntime tests * trigger slow * don't specify gpu for tpu * optimize * memory limit * fix flax tests * disable docker cache
42 lines
1.2 KiB
Docker
42 lines
1.2 KiB
Docker
FROM nvidia/cuda:11.6.2-cudnn8-devel-ubuntu20.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 \
|
|
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 \
|
|
"onnxruntime-gpu>=1.13.1" \
|
|
--extra-index-url https://download.pytorch.org/whl/cu117 && \
|
|
python3 -m pip install --no-cache-dir \
|
|
accelerate \
|
|
datasets \
|
|
hf-doc-builder \
|
|
huggingface-hub \
|
|
modelcards \
|
|
numpy \
|
|
scipy \
|
|
tensorboard \
|
|
transformers
|
|
|
|
CMD ["/bin/bash"] |