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

RUN apt-get update
RUN apt-get install -y git

COPY requirements.txt /tmp/requirements.txt

RUN pip install -r /tmp/requirements.txt