Update Dockerfile

borchero · borchero · commit be2d7852c8c7 · 2020-04-05T23:51:32.000+02:00
diff --git a/Dockerfile b/Dockerfile
@@ -2,5 +2,8 @@ FROM python:3.7
 
 RUN pip install pylint twine sphinx sphinx-rtd-theme
 
+# Cache PyTorch download
+RUN pip install "torch>=1.4.0,<2.0.0"
+
 COPY requirements.txt /requirements.txt
 RUN pip install -r requirements.txt
