diff --git a/build-docker.sh b/build-docker.sh
index 17072693f..d4d865a95 100755
--- a/build-docker.sh
+++ b/build-docker.sh
@@ -16,7 +16,7 @@ key="$1"
 case $key in
     --gpu)
     GPU="-gpu"
-    BASE_IMAGE="nvidia/cuda:10.1-cudnn7d-runtime-ubuntu18.04"
+    BASE_IMAGE="nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04"
     ;;
     --no-cache-build)
     NO_CACHE="--no-cache"