declare-lab/tango2

1FROM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu20.042LABEL maintainer="Hugging Face"3LABEL repository="diffusers"4 5ENV DEBIAN_FRONTEND=noninteractive6 7RUN apt update && \8    apt install -y bash \9                   build-essential \10                   git \11                   git-lfs \12                   curl \13                   ca-certificates \14                   libsndfile1-dev \15                   python3.8 \16                   python3-pip \17                   python3.8-venv && \18    rm -rf /var/lib/apt/lists19 20# make sure to use venv21RUN python3 -m venv /opt/venv22ENV PATH="/opt/venv/bin:$PATH"23 24# pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)25RUN python3 -m pip install --no-cache-dir --upgrade pip && \26    python3 -m pip install --no-cache-dir \27        torch \28        torchvision \29        torchaudio \30    python3 -m pip install --no-cache-dir \31        accelerate \32        datasets \33        hf-doc-builder \34        huggingface-hub \35        Jinja2 \36        librosa \37        numpy \38        scipy \39        tensorboard \40        transformers41 42CMD ["/bin/bash"]43