From 8126b73488bfad83d4c3177e9d7566fd91348923 Mon Sep 17 00:00:00 2001 From: "Huang, Guangtai" Date: Sat, 11 Sep 2021 00:32:43 +0800 Subject: [PATCH] [Docker] Fix apex installation (#755) --- docker/Dockerfile.ci_cpu | 2 +- docker/Dockerfile.ci_gpu | 2 +- docker/install/ubuntu_install_python.sh | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/Dockerfile.ci_cpu b/docker/Dockerfile.ci_cpu index 184773a5..6d30f107 100644 --- a/docker/Dockerfile.ci_cpu +++ b/docker/Dockerfile.ci_cpu @@ -7,7 +7,7 @@ COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh RUN bash /install/ubuntu_install_core.sh COPY install/ubuntu_install_python.sh /install/ubuntu_install_python.sh -RUN bash /install/ubuntu_install_python.sh +RUN bash /install/ubuntu_install_python.sh cpu COPY install/ubuntu_install_llvm.sh /install/ubuntu_install_llvm.sh RUN bash /install/ubuntu_install_llvm.sh diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index 66286bb6..b82982af 100644 --- a/docker/Dockerfile.ci_gpu +++ b/docker/Dockerfile.ci_gpu @@ -7,7 +7,7 @@ COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh RUN bash /install/ubuntu_install_core.sh COPY install/ubuntu_install_python.sh /install/ubuntu_install_python.sh -RUN bash /install/ubuntu_install_python.sh +RUN bash /install/ubuntu_install_python.sh gpu COPY install/ubuntu_install_llvm.sh /install/ubuntu_install_llvm.sh RUN bash /install/ubuntu_install_llvm.sh diff --git a/docker/install/ubuntu_install_python.sh b/docker/install/ubuntu_install_python.sh index b278bb5b..01caa9da 100755 --- a/docker/install/ubuntu_install_python.sh +++ b/docker/install/ubuntu_install_python.sh @@ -34,7 +34,7 @@ pip3 install torch==1.8.1+cu102 torchvision==0.9.1+cu102 -f https://download.pyt pip3 install mxnet==1.6.0 pip3 install gluoncv==0.10.1 -if [ -x "$(command -v nvidia-smi)" ]; then +if [[ "$1" == "gpu" ]]; then mkdir -p build && cd build git clone https://github.com/szhengac/apex --branch lans cd apex