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"