From 7d015b6da3d75ec38c3badaebc5b58e58dafae33 Mon Sep 17 00:00:00 2001
From: Christoph Schranz <christoph.schranz@salzburgresearch.at>
Date: Sun, 23 Feb 2020 19:01:07 +0100
Subject: [PATCH] change to cuda 10.2

---
 .build/Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.build/Dockerfile b/.build/Dockerfile
index c007c06..0328c86 100644
--- a/.build/Dockerfile
+++ b/.build/Dockerfile
@@ -1,6 +1,6 @@
 # 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.1-base-ubuntu18.04
+FROM nvidia/cuda:10.2-base-ubuntu18.04
 LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"
 
 ############################################################################
@@ -380,7 +380,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.1 \
+    cudatoolkit=10.2 \
     'pytorch=1.3.1' \
     torchvision && \
     conda clean --all -f -y && \