From e16818d5f553b47d00bdc6427e67a0fd82342937 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 | 3 +++ ci/docker/Dockerfile.build.ubuntu_blc | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/cd/utils/mxnet_base_image.sh b/cd/utils/mxnet_base_image.sh index 5073a3783b2c..5632fe8d024a 100755 --- a/cd/utils/mxnet_base_image.sh +++ b/cd/utils/mxnet_base_image.sh @@ -33,6 +33,9 @@ case ${mxnet_variant} in cu110*) echo "nvidia/cuda:11.0-cudnn8-runtime-ubuntu18.04" ;; + cu112*) + echo "nvidia/cuda:11.2.1-cudnn8-runtime-ubuntu18.04" + ;; cpu) echo "ubuntu:18.04" ;; diff --git a/ci/docker/Dockerfile.build.ubuntu_blc b/ci/docker/Dockerfile.build.ubuntu_blc index a413648366b2..7c1054d637dc 100644 --- a/ci/docker/Dockerfile.build.ubuntu_blc +++ b/ci/docker/Dockerfile.build.ubuntu_blc @@ -18,7 +18,7 @@ # # Dockerfile to build and run MXNet on Ubuntu 16.04 for CPU -FROM ubuntu:16.04 +FROM nvidia/cuda:11.2.1-cudnn8-devel-ubuntu16.04 WORKDIR /work/deps