rebase to nvidia/cuda:10.1-base-ubuntu18.04 as 10.2 makes problems

iot-salzburg · Feb 22, 2020 · 5b54ba1 · 5b54ba1
1 parent 6a0dd56
commit 5b54ba1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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 <[email protected]>"
diff --git 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 && \