FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime

RUN <<EOF
apt-get update
apt-get install -y --no-install-recommends git
rm -rf /var/lib/apt/lists/*
EOF

COPY requirements.txt requirements.txt

RUN pip install --no-cache-dir -r requirements.txt