FROM nvidia/cuda:12.6.0-cudnn-devel-ubuntu24.04

ENV DEBIAN_FRONTEND=noninteractive
ENV PIP_NO_CACHE_DIR=1
ENV PYTHONUNBUFFERED=1
ENV VIRTUAL_ENV=/opt/venv
ENV PATH="/opt/venv/bin:${PATH}"

RUN apt-get update && apt-get install -y --no-install-recommends \
    ca-certificates \
    python3 \
    python3-pip \
    python3-venv \
    && rm -rf /var/lib/apt/lists/*

RUN python3 -m venv "${VIRTUAL_ENV}"

WORKDIR /workspace

COPY requirements.txt pyproject.toml README.md LICENSE ./
COPY spacy_accelerate ./spacy_accelerate
COPY benchmarks ./benchmarks

RUN python -m pip install --upgrade pip setuptools wheel \
    && python -m pip install -r requirements.txt \
    && python -m pip install -e . --no-deps \
    && python -m pip install datasets tqdm pynvml psutil orjson \
    && python -m spacy download en_core_web_trf \
    && chmod +x /workspace/benchmarks/docker/run.sh

ENTRYPOINT ["/workspace/benchmarks/docker/run.sh"]
