From f76928aea3f9f229af0be3b8f5c375407e16b907 Mon Sep 17 00:00:00 2001 From: Rohit Kumar Srivastava Date: Thu, 25 Feb 2021 23:56:27 +0000 Subject: [PATCH] correcting cuda 11.2 image name in CI and CD --- 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