FROM        pytorch/pytorch:1.7.0-cuda11.0-cudnn8-runtime
MAINTAINER  nawnoes@gmail.com

COPY        ./requirements.txt /usr/src/requirements.txt
RUN         pip install -r /usr/src/requirements.txt
