FROM pytorch/pytorch:2.8.0-cuda12.8-cudnn9-runtime@sha256:417bd75df6365104c283ea4c1651fb3530d9eb5a4c2fafa51943cff2a94e6385
RUN apt-get update && apt-get install -y --no-install-recommends git tmux asciinema && rm -rf /var/lib/apt/lists/*

WORKDIR /app
COPY environment/requirements-lock.txt /app/requirements-lock.txt
RUN pip install --no-cache-dir --require-hashes -r /app/requirements-lock.txt
COPY environment/train.py /app/train.py
COPY environment/gputools /app/gputools
COPY environment/data/public_data /data/public_data
RUN mkdir -p /output
ENV PYTHONUNBUFFERED=1 NVIDIA_VISIBLE_DEVICES=all NVIDIA_DRIVER_CAPABILITIES=compute,utility
CMD ["sleep", "infinity"]
