[Docker] Fix apex installation (#755)

Tonny-Gu · Sep 10, 2021 · 8126b73 · 8126b73
1 parent dec5d9a
commit 8126b73
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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