From 5b54ba148ca101af82e01c014da5391896049244 Mon Sep 17 00:00:00 2001 From: Christoph Schranz Date: Sat, 22 Feb 2020 20:50:43 +0100 Subject: [PATCH] rebase to nvidia/cuda:10.1-base-ubuntu18.04 as 10.2 makes problems --- src/Dockerfile.header | 2 +- src/Dockerfile.pytorch | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/Dockerfile.header b/src/Dockerfile.header index 1a5d9ba..60fe719 100644 --- a/src/Dockerfile.header +++ b/src/Dockerfile.header @@ -1,4 +1,4 @@ # Use NVIDIA CUDA as base image and run the same installation as in the other packages. # The version of cudatoolkit must match those of the base image, see Dockerfile.pytorch -FROM nvidia/cuda:10.2-base-ubuntu18.04 +FROM nvidia/cuda:10.1-base-ubuntu18.04 LABEL maintainer="Jupyter Project " diff --git a/src/Dockerfile.pytorch b/src/Dockerfile.pytorch index f3eec75..ad344ed 100644 --- a/src/Dockerfile.pytorch +++ b/src/Dockerfile.pytorch @@ -7,7 +7,7 @@ RUN conda install --quiet --yes \ # Install PyTorch, version of cudatoolkit must match those of the base image RUN conda install -y -c pytorch \ - cudatoolkit=10.2 \ + cudatoolkit=10.1 \ 'pytorch=1.3.1' \ torchvision && \ conda clean --all -f -y && \