Update Dockerfile

borchero · Apr 5, 2020 · be2d785 · be2d785
1 parent fbe121a
commit be2d785
Showing 1 changed file with 3 additions and 0 deletions.
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