From be2d7852c8c75db55cfe7ba39b84b3b554ecd8e3 Mon Sep 17 00:00:00 2001 From: Oliver Borchert Date: Sun, 5 Apr 2020 23:51:32 +0200 Subject: [PATCH] Update Dockerfile --- Dockerfile | 3 +++ 1 file changed, 3 insertions(+) diff --git a/Dockerfile b/Dockerfile index 2f525c4..48fb931 100644 --- 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