Dockerfile

FROM nvidia/cuda:11.2.0-cudnn8-runtime-ubuntu20.04

RUN apt-get update -y && DEBIAN_FRONTEND=noninteractive apt-get install -y \
    # Install python 3.8, and other tools if needed:
    # Note: Since we are in a Ubunut 20.04 image, this installs python-3.8
    python3 python3-pip python3-venv wget \
    # Install audio dependencies if needed:
    # ffmpeg libsndfile-dev \
    && apt-get clean && rm -rf /var/lib/apt/lists/*

# Turn off buffering for easier container logging
ENV PYTHONUNBUFFERED 1

# Configure Poetry
ENV POETRY_HOME=/opt/poetry
ENV POETRY_VENV=/opt/poetry-venv
ENV POETRY_CACHE_DIR=/opt/.cache

# Install Poetry separated from system interpreter
RUN python3 -m venv $POETRY_VENV \
    && $POETRY_VENV/bin/pip install -U pip setuptools \
    && $POETRY_VENV/bin/pip install poetry==1.3.0
# Add `poetry` to PATH
ENV PATH="${PATH}:${POETRY_VENV}/bin"

# download model
RUN wget -nc https://github.com/deezer/pauzee_taln23/releases/download/v1.0.0/model_pauzee.tar.gz
RUN tar -zxf model_pauzee.tar.gz -C .
RUN rm -r model_pauzee.tar.gz

# Now we can install the project's Python dependencies using Poetry
WORKDIR /workspace
COPY pyproject.toml poetry.lock ./
RUN poetry install --no-root

# Copy the project code and dataset
# COPY code/ code/
# COPY data/ data/
RUN mkdir -p models/
RUN cp -r /models/* models/