From 006856b9a11349ce5513a090a7ad7f3d9d2282cc Mon Sep 17 00:00:00 2001
From: Lee moon soo <leemoonsoo@gmail.com>
Date: Thu, 10 Dec 2020 14:31:59 -0800
Subject: [PATCH] fix gpu base image name in build-docker.sh script (#12642)

---
 build-docker.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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"