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

RUN apt-get update
RUN apt-get install -y git
# RUN apt-get install texlive-latex-base texlive-fonts-recommended texlive-fonts-extra texlive-bibtex-extra

COPY requirements.txt /tmp/requirements.txt

RUN pip install -r /tmp/requirements.txt