[incubator-mxnet] branch v1.9.1-test updated: update key only for gpu

[we...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

weichu pushed a commit to branch v1.9.1-test
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/v1.9.1-test by this push:
     new a0629c21d5 update key only for gpu
a0629c21d5 is described below

commit a0629c21d5b7403868bb0aefacf5310cb5fcd9e3
Author: Wei Chu <we...@amazon.com>
AuthorDate: Mon May 16 14:06:54 2022 -0700

    update key only for gpu
---
 cd/python/docker/Dockerfile | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/cd/python/docker/Dockerfile b/cd/python/docker/Dockerfile
index 76275c3b8b..7314f795dc 100644
--- a/cd/python/docker/Dockerfile
+++ b/cd/python/docker/Dockerfile
@@ -23,8 +23,12 @@
 ARG BASE_IMAGE
 FROM ${BASE_IMAGE}
 
+ARG MXNET_VARIANT
+
 # update cuda repo key: https://developer.nvidia.com/blog/updating-the-cuda-linux-gpg-repository-key/
-RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
+RUN if [[ $MXNET_VARIANT = cu* ]] ; then \
+    apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub; \
+    fi
 
 RUN apt-get update && \
     apt-get install -y software-properties-common && \
@@ -37,7 +41,6 @@ RUN apt-get update && \
 
 RUN apt-get install -y libgomp1
 
-ARG MXNET_VARIANT
 RUN if [ "$MXNET_VARIANT" = "aarch64_cpu" ] ; then echo "not installing libquadmath0 on aarch64" ; else apt-get install -y libquadmath0 ; fi
 RUN if [ "$MXNET_VARIANT" = "aarch64_cpu" ] ; then \
     wget https://armkeil.blob.core.windows.net/developer/Files/downloads/hpc/arm-performance-libraries/21-0-0/Ubuntu18.04/arm-performance-libraries_21.0_Ubuntu-18.04_gcc-8.2.tar && \