Dockerfile

FROM pytorch/pytorch:latest

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

# install requirements
COPY requirements.txt .
RUN pip install -r requirements.txt