@@ -55,30 +55,32 @@ RUN source /opt/utils/script-utils.sh \
5555 && ( which java && echo "py4j % Install py4j if Java exists" >> /opt/utils/install_list_PY_datascience.pip || echo "Skip py4j install" ) \
5656 ) || echo "Skip Python datascience packages install" ) \
5757 && for profile in $(echo $ARG_PROFILE_PYTHON | tr "," "\n " ) ; do ( echo "Pip install for ${profile}" ; install_pip "/opt/utils/install_list_PY_${profile}.pip" ; ) ; done \
58- # Handle tensorflow installation 1.x/2.x, cpu/gpu: https://www.tensorflow.org/install/source#gpu
58+ # Handle tensorflow installation 1.x/2.x, cpu/gpu: https://www.tensorflow.org/install/source#gpu
5959 && ( $(grep -q "tf" <<< "${ARG_PROFILE_PYTHON}" ) && ( \
6060 TF=$( [ -x "$(command -v nvcc)" ] && echo "tensorflow-gpu" || echo "tensorflow" ) \
6161 && V=$($(grep -q "tf1" <<< "${ARG_PROFILE_PYTHON}" ) && echo "1" || echo "2" ) \
6262 && pip install --pre -U "${TF}==${V}.*" \
6363 ) || echo "Skip tf install" ) \
64- # Handle pytorch installation 1.x only, cpu/gpu: https://pytorch.org/get-started/locally/
64+ # Handle pytorch installation 1.x only, cpu/gpu: https://pytorch.org/get-started/locally/
6565 && ( $(grep -q "torch" <<< "${ARG_PROFILE_PYTHON}" ) && ( \
6666 CUDA_VER=$(echo "${CUDA_VERSION:0:4}" | sed 's/\. //' ) \
6767 && IDX=$( [ -x "$(command -v nvcc)" ] && echo "cu${CUDA_VER:-113}" || echo "cpu" ) \
6868 && pip install --pre -U torch -f "https://download.pytorch.org/whl/${IDX}/torch_stable.html" \
6969 ) || echo "Skip pytorch install" ) \
70- # Handle paddle installation, cpu/gpu: https://www.paddlepaddle.org.cn/
70+ # Handle paddle installation, cpu/gpu: https://www.paddlepaddle.org.cn/
71+ # URL_PADDLE=$(curl -sL "https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html" |tr "<>" "\n" | grep "${PADDLE_VER}" | grep href | cut -d '=' -f 2)
7172 && ( $(grep -q "paddle" <<< "${ARG_PROFILE_PYTHON}" ) && ( \
7273 PADDLE=$( [ -x "$(command -v nvcc)" ] && echo "paddlepaddle-gpu" || echo "paddlepaddle" ) \
7374 && PADDLE_VER=$(pip index versions ${PADDLE} | grep '(' | cut -d "(" -f 2 | tr -d ')[:space:]' ) \
7475 && CUDA_VER=$(echo "${CUDA_VERSION:0:4}" | sed 's/\. //' ) \
7576 && V=$($(grep -q "11" <<< "${CUDA_VER}" ) && echo ".post${CUDA_VER}" || echo "" ) \
76- && V=$(echo ${PADDLE}==${PADDLE_VER}${V}) && echo "paddle to install: ${V}" \
77+ && V=$(echo ${PADDLE}==${PADDLE_VER}${V}) && echo "to install paddle: ${V}" \
78+ && pip instal pip==20.2.2 \
7779 && pip install ${V} -f "https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html" \
80+ && pip install -U pip \
7881 ) || echo "Skip paddle install" )
7982
8083
81-
8284RUN $(grep -q "base" <<< "${ARG_PROFILE_GO}" ) && ( source /opt/utils/script-setup.sh && setup_GO ) || echo "Skip go install"
8385
8486RUN $(grep -q "base" <<< "${ARG_PROFILE_JULIA}" ) && ( source /opt/utils/script-setup.sh && setup_julia ) || echo "Skip julia install"
0 commit comments