From 85d07a33423ddc861a8ffee02264078f165d7382 Mon Sep 17 00:00:00 2001
From: "xingjun.wxj" <xingjun.wxj@alibaba-inc.com>
Date: Fri, 21 Mar 2025 10:17:27 +0800
Subject: [PATCH] fix installing apex in gpu base image

---
 docker/Dockerfile.ubuntu_base  | 5 +++--
 docker/scripts/install_apex.sh | 1 -
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/docker/Dockerfile.ubuntu_base b/docker/Dockerfile.ubuntu_base
index 07096766..0816e868 100644
--- a/docker/Dockerfile.ubuntu_base
+++ b/docker/Dockerfile.ubuntu_base
@@ -186,14 +186,15 @@ RUN if [ "$USE_GPU" = "True" ] ; then \
 RUN if [ "$USE_GPU" = "True" ] ; then \
        export TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.9;9.0;8.6+PTX" && pip install --no-cache-dir git+https://github.com/gxd1994/Pointnet2.PyTorch.git@master#subdirectory=pointnet2; \
     else \
-     echo 'cpu unsupport Pointnet2'; \
+     echo 'CPU env does not support Pointnet2'; \
     fi
 
 # NVIDIA apex building
 RUN if [ "$USE_GPU" = "True" ] ; then \
+        pip install --no-cache-dir packaging &&\
         bash /tmp/install_apex.sh; \
     else \
-     echo 'cpu unsupport nvidia apex'; \
+     echo 'CPU env does not support NVIDIA Apex.'; \
     fi
 
 
diff --git a/docker/scripts/install_apex.sh b/docker/scripts/install_apex.sh
index 8f2fa39f..7ecd288b 100644
--- a/docker/scripts/install_apex.sh
+++ b/docker/scripts/install_apex.sh
@@ -3,6 +3,5 @@ export MAX_JOBS=16 \
 && cd apex \
 && git checkout 6bd01c4b99a84648ad5e5238a959735e6936c813 \
 && TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5;8.0;8.9;9.0;8.6+PTX" pip install -v --disable-pip-version-check --no-cache --global-option="--cpp_ext" --global-option="--cuda_ext" ./ \
-&& pip install packaging \
 && cd .. \
 && rm -rf apex