fix python version

Author: haobibo

.github/workflows/docker.yml

@@ -234,7 +234,7 @@ jobs:
       - run: |
           source ./tool.sh
           build_image py-cuda-10.0 latest docker_core/Dockerfile \
-            --build-arg "ARG_PROFILE_PYTHON=tf1,datascience,nlp,cv" \
+            --build-arg "ARG_PROFILE_PYTHON=tf1" \
             --build-arg "BASE_IMG=cuda_10.0"
           alias_image py-cuda-10.0 latest tf1 latest && push_image
 
@@ -247,7 +247,7 @@ jobs:
       - run: |
           source ./tool.sh
           build_image py-cuda-11.2 latest docker_core/Dockerfile \
-            --build-arg "ARG_PROFILE_PYTHON=tf2,datascience,nlp,cv" \
+            --build-arg "ARG_PROFILE_PYTHON=tf2" \
             --build-arg "BASE_IMG=cuda_11.2"
           alias_image py-cuda-11.2 latest tf2 latest && push_image
 
@@ -260,7 +260,7 @@ jobs:
       - run: |
           source ./tool.sh
           build_image py-cuda-11.3 latest docker_core/Dockerfile \
-            --build-arg "ARG_PROFILE_PYTHON=torch,datascience,nlp,cv" \
+            --build-arg "ARG_PROFILE_PYTHON=torch" \
             --build-arg "BASE_IMG=cuda_11.3"
           alias_image py-cuda-11.3 latest torch latest && push_image
 

docker_atom/work/script-setup.sh

@@ -33,7 +33,8 @@ setup_conda_postprocess() {
 }
 
 setup_conda_with_mamba() {
-  mamba install -y --root-prefix="${CONDA_PREFIX}" --prefix="${CONDA_PREFIX}" conda pip -c "conda-forge/linux-64"
+  VERSION_PTYHON=$1; shift 1;
+  mamba install -y --root-prefix="${CONDA_PREFIX}" --prefix="${CONDA_PREFIX}" -c "conda-forge" conda pip python="${VERSION_PTYHON:-3.9}"
   rm -rf /opt/conda/pkgs/*
   setup_conda_postprocess
 }

docker_core/Dockerfile

@@ -62,7 +62,7 @@ RUN source /opt/utils/script-utils.sh \
       || true ) \
  # Handle pytorch installation 1.x only, cpu/gpu: https://pytorch.org/get-started/locally/
  && ( [[ ${ARG_PROFILE_PYTHON} == *"torch"* ]] \
-      && CUDA_VER=$(echo "${NVIDIA_REQUIRE_CUDA/./}" | cut -d'=' -f 2) \
+      && CUDA_VER=$(echo "${CUDA_VERSION:0:4}" | sed 's/\.//' ) \
       && IDX=$( [ -x "$(command -v nvcc)" ] && echo "cu${CUDA_VER:-113}" || echo "cpu" ) \
       && pip install --pre -U torch -f "https://download.pytorch.org/whl/${IDX}/torch_stable.html" \
       || true )