FROM python@sha256:6e13e65c55e33adf203d77ee371cf8bf5d81bd4902ef07565721f46bf44917af
WORKDIR /app
ENV PYTHONDONTWRITEBYTECODE=1 PYTHONUNBUFFERED=1 HOME=/tmp TOKENIZERS_PARALLELISM=false
ENV USER=reader LOGNAME=reader TORCHINDUCTOR_CACHE_DIR=/tmp/torchinductor
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
ENV LD_LIBRARY_PATH=/usr/local/lib/python3.12/site-packages/nvidia/cublas/lib:/usr/local/lib/python3.12/site-packages/nvidia/cudnn/lib
COPY requirements.lock.txt /app/requirements.lock.txt
RUN python -m pip install --no-cache-dir --require-hashes -r requirements.lock.txt && python -m pip check
RUN python -m pip freeze > /app/environment.freeze.txt
COPY . /app
ENTRYPOINT ["python"]
