You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by zh...@apache.org on 2021/02/26 18:22:24 UTC
[incubator-mxnet] branch v1.x updated: correcting cuda 11.2 image
name in CI and CD (#19960)
This is an automated email from the ASF dual-hosted git repository.
zha0q1 pushed a commit to branch v1.x
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/v1.x by this push:
new a9eec24 correcting cuda 11.2 image name in CI and CD (#19960)
a9eec24 is described below
commit a9eec24a773fd068cb29e83f111f8c9cfa405768
Author: Rohit Kumar Srivastava <sr...@osu.edu>
AuthorDate: Fri Feb 26 10:18:54 2021 -0800
correcting cuda 11.2 image name in CI and CD (#19960)
Co-authored-by: Rohit Kumar Srivastava <sr...@buckeyemail.osu.edu>
---
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 e2ea39d..3af6869 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 1470673..a6e479f 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