From a9eec24a773fd068cb29e83f111f8c9cfa405768 Mon Sep 17 00:00:00 2001 From: Rohit Kumar Srivastava Date: Fri, 26 Feb 2021 10:18:54 -0800 Subject: [PATCH] correcting cuda 11.2 image name in CI and CD (#19960) Co-authored-by: Rohit Kumar Srivastava --- cd/utils/mxnet_base_image.sh | 2 +- ci/docker/Dockerfile.build.ubuntu_gpu_cu112 | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cd/utils/mxnet_base_image.sh b/cd/utils/mxnet_base_image.sh index e2ea39d404be..3af6869bd915 100755 --- a/cd/utils/mxnet_base_image.sh +++ b/cd/utils/mxnet_base_image.sh @@ -34,7 +34,7 @@ case ${mxnet_variant} in echo "nvidia/cuda:11.0-cudnn8-runtime-ubuntu16.04" ;; cu112*) - echo "nvidia/cuda:11.2-cudnn8-runtime-ubuntu16.04" + echo "nvidia/cuda:11.2.1-cudnn8-runtime-ubuntu16.04" ;; cpu) echo "ubuntu:16.04" diff --git a/ci/docker/Dockerfile.build.ubuntu_gpu_cu112 b/ci/docker/Dockerfile.build.ubuntu_gpu_cu112 index 14706737620b..a6e479f57a13 100644 --- a/ci/docker/Dockerfile.build.ubuntu_gpu_cu112 +++ b/ci/docker/Dockerfile.build.ubuntu_gpu_cu112 @@ -18,7 +18,7 @@ # # Dockerfile to run MXNet on Ubuntu 16.04 for GPU -FROM nvidia/cuda:11.2-cudnn8-devel-ubuntu16.04 +FROM nvidia/cuda:11.2.1-cudnn8-devel-ubuntu16.04 WORKDIR /work/deps