Dockerfile

# Canonical environment of ENVIRONMENT.md: Python 3.12.3 + requirements.lock (torch 2.14.0 from PyPI, built for
# CUDA 13.0; the CUDA libraries come as pip wheels and the NVIDIA driver from the host). The model is not copied
# into the image: mount this repository at /model.
#
#   docker build -t ines-1-env .
#   docker run --rm --gpus all -v "$PWD":/model ines-1-env                 # GPU (bf16 + autocast)
#   docker run --rm -v "$PWD":/model ines-1-env                            # CPU (fp32)
FROM python:3.12.3-slim-bookworm
ENV PIP_NO_CACHE_DIR=1 PYTHONDONTWRITEBYTECODE=1
COPY requirements.lock /tmp/requirements.lock
RUN pip install -r /tmp/requirements.lock
WORKDIR /model
CMD ["python", "examples/quickstart.py"]