/
ai-forever
/
diffusers
Обзор
Документация
Войти
/
ai-forever
/
diffusers
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
main
docker/diffusers-pytorch-cuda/Dockerfile
46 строк
1017 B
Dhruv Nair
Fix Slow Tests (#5469)
23 окт 2023, 21:24
Не верифицирован
23 окт 2023, 21:24
33293ed
Код
Авторство
О чём код?
FROM nvidia/cuda:12.1.0-runtime-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 \ libsndfile1-dev \ libgl1 \ 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 \ invisible_watermark && \ python3 -m pip install --no-cache-dir \ accelerate \ datasets \ hf-doc-builder \ huggingface-hub \ Jinja2 \ librosa \ numpy \ scipy \ tensorboard \ transformers \ omegaconf \ pytorch-lightning CMD ["/bin/bash"]