diff --git a/src/Dockerfile.header b/src/Dockerfile.header
index 60fe719..1a5d9ba 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.1-base-ubuntu18.04
+FROM nvidia/cuda:10.2-base-ubuntu18.04
 LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"
diff --git a/src/Dockerfile.pytorch b/src/Dockerfile.pytorch
index ad344ed..f3eec75 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.1 \
+    cudatoolkit=10.2 \
     'pytorch=1.3.1' \
     torchvision && \
     conda clean --all -f -y && \