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
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
COPY training.lock.txt /app/training.lock.txt
RUN python -m pip install --no-cache-dir --require-hashes -r training.lock.txt && python -m pip check
RUN python -m pip freeze > /app/environment.freeze.txt
COPY . /app
ENTRYPOINT ["python"]
