2024-02-29 08:22:53 -06:00

12 lines
246 B
Docker

FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-devel
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