diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index 87ce9e8..ba8e04e 100644
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:10.2-cudnn8-runtime-ubuntu18.04 AS python-ubuntu
+FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04 AS python-ubuntu
 
 RUN apt-get update
 RUN apt-get install -y build-essential curl python3 python3-pip python3-venv