FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime

COPY requirements.txt /tmp/requirements.txt

RUN pip install -r /tmp/requirements.txt