ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:25.04-py3-igpu
FROM ${BASE_IMAGE}

COPY --from=ghcr.io/astral-sh/uv:0.8.13 /uv /uvx /bin/

RUN uv venv --system-site-packages /opt/nomos-vlm

ENV PATH="/opt/nomos-vlm/bin:${PATH}"

RUN uv pip install --python /opt/nomos-vlm/bin/python --no-cache --no-deps \
    "accelerate==1.9.0" \
    "docopt==0.6.2" \
    "hf-xet==1.1.5" \
    "huggingface-hub==0.33.4" \
    "num2words==0.5.14" \
    "safetensors==0.5.3" \
    "sentencepiece==0.2.0" \
    "tokenizers==0.21.2" \
    "transformers==4.53.2"

RUN python -c \
    'import accelerate, huggingface_hub, num2words, safetensors, tokenizers, torch, transformers; assert ".nv25.04" in torch.__version__'

ENV PYTHONUNBUFFERED=1
