FROM nvcr.io/nvidia/pytorch:25.11-py3
WORKDIR /workspace
COPY requirements-notebook.txt /tmp/requirements-notebook.txt
RUN python -m pip install --no-cache-dir -r /tmp/requirements-notebook.txt
COPY . /workspace
CMD ["python", "experiment.py", "--device", "cuda"]
