fix setuptool version

Author: zhuofuAMZ

base/buildspec.yml

@@ -2,8 +2,8 @@ account_id: &ACCOUNT_ID <set-$ACCOUNT_ID-in-environment>
 prod_account_id: &PROD_ACCOUNT_ID 763104351884
 region: &REGION <set-$REGION-in-environment>
 framework: &FRAMEWORK base
-version: &VERSION 12.8.1
-short_version: &SHORT_VERSION "12.8"
+version: &VERSION 12.6.3
+short_version: &SHORT_VERSION "12.6"
 arch_type: &ARCH_TYPE x86_64
 autopatch_build: "False"
 
@@ -32,7 +32,7 @@ context:
       target: install_efa.sh
 
 images:
-  base_x86_64_gpu_cuda128:
+  base_x86_64_gpu_cuda126:
     <<: *BASE_REPOSITORY
     context:
       <<: *BASE_CONTEXT

base/x86_64/gpu/cu128/Dockerfile

@@ -97,9 +97,7 @@ COPY --from=cuda-builder /usr/local/cuda-${CUDA_MAJOR}.${CUDA_MINOR} /usr/local/
 COPY install_efa.sh install_efa.sh 
 COPY deep_learning_container.py /usr/local/bin/deep_learning_container.py
 COPY bash_telemetry.sh /usr/local/bin/bash_telemetry.sh
-RUN ${PIP} --no-cache-dir install --upgrade \
-   pip \
-   setuptools
+
 RUN chmod +x /usr/local/bin/deep_learning_container.py && \
     chmod +x /usr/local/bin/bash_telemetry.sh && \
     echo 'source /usr/local/bin/bash_telemetry.sh' >> /etc/bash.bashrc && \

scripts/install_python.sh

@@ -24,6 +24,7 @@ function install_python {
     # this will add pip systemlink to pip${PYTHON_MAJOR_VERSION}
     python -m pip install --upgrade pip --trusted-host pypi.org --trusted-host files.pythonhosted.org
     python -m pip install --no-cache-dir awscli boto3 requests setuptools
+    python -m pip install --no-cache-dir --upgrade  setuptools
 }
 
 # idiomatic parameter and option handling in sh