From 2f9b5d8acd69f741b0d5f852d6d8644f0a3f4e74 Mon Sep 17 00:00:00 2001 From: zha0q1 Date: Mon, 5 Apr 2021 18:46:30 +0000 Subject: [PATCH] attemp to fix nightly docker for master cu112 --- cd/utils/mxnet_base_image.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cd/utils/mxnet_base_image.sh b/cd/utils/mxnet_base_image.sh index 87a70c9343cb..1701a884a77f 100755 --- a/cd/utils/mxnet_base_image.sh +++ b/cd/utils/mxnet_base_image.sh @@ -31,7 +31,7 @@ case ${mxnet_variant} in echo "nvidia/cuda:11.0-cudnn8-runtime-ubuntu18.04" ;; cu112*) - echo "nvidia/cuda:11.2.1-cudnn8-runtime-ubuntu18.04" + echo "nvidia/cuda:11.2.0-cudnn8-runtime-ubuntu18.04" ;; cpu) echo "ubuntu:18.04"