fix gpu base image name in build-docker.sh script (#12642)

This commit is contained in:
Lee moon soo
2020-12-10 14:31:59 -08:00
committed by GitHub
parent 932837eb4c
commit 006856b9a1
+1 -1
View File
@@ -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"