Dockerfile

# The builder image, used to build the virtual environment
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime

RUN pip install poetry==2.0.1

ENV POETRY_NO_INTERACTION=1 \
    POETRY_VIRTUALENVS_IN_PROJECT=1 \
    POETRY_VIRTUALENVS_CREATE=1 \
    POETRY_CACHE_DIR=/tmp/poetry_cache \
    PYTHONUNBUFFERED=1

WORKDIR /app

COPY pyproject.toml poetry.lock ./
#not really needed, but poetry seems to want it
RUN touch README.md 


RUN poetry cache clear . --all
RUN poetry install --no-root && rm -rf $POETRY_CACHE_DIR

WORKDIR /app
ENV VIRTUAL_ENV=/app/.venv \
    PATH="/app/.venv/bin:$PATH"

COPY main.py ./main.py
COPY config.yaml ./
COPY src ./src
COPY data ./data
COPY model ./model

CMD python main.py