diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 000000000..3e6da2209 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,13 @@ +FROM pytorch/pytorch:1.5-cuda10.1-cudnn7-runtime + +ENV DEBIAN_FRONTEND=noninteractive + +RUN apt-get update -qq && \ + apt-get install -y git vim libgtk2.0-dev && \ + rm -rf /var/cache/apk/* + +RUN pip --no-cache-dir install Cython + +COPY requirements.txt /workspace + +RUN pip --no-cache-dir install -r /workspace/requirements.txt