@@ -35,13 +35,13 @@ function prepare_torch_env() {
3535 LOG " [INFO] Device Id: ${CUDA_VISIBLE_DEVICES} "
3636 # Update pip
3737 LOG " [INFO] Update pip ..."
38- env http_proxy=" " https_proxy=" " pip install -U pip > /dev/null
38+ env http_proxy=" " https_proxy=" " python3.10 -m pip install -U pip > /dev/null
3939 [ $? -ne 0 ] && LOG " [FATAL] Update pip failed!" && exit -1
4040 # install torch
41- pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu126 > /dev/null
41+ python3.10 -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu126 > /dev/null
4242 [ $? -ne 0 ] && LOG " [FATAL] Install torch2.9.0 failed!" && exit -1
4343 else
44- python ${GRAPH_NET_EXTRACT_WORKSPACE} /tools/count_sample.py
44+ python3.10 ${GRAPH_NET_EXTRACT_WORKSPACE} /tools/count_sample.py
4545 LOG " [INFO] This pull request doesn't change any torch samples, skip the CI."
4646 fi
4747}
@@ -53,16 +53,16 @@ function prepare_paddle_env() {
5353 LOG " [INFO] Device Id: ${CUDA_VISIBLE_DEVICES} "
5454 # Update pip
5555 LOG " [INFO] Update pip ..."
56- env http_proxy=" " https_proxy=" " pip install -U pip > /dev/null
56+ env http_proxy=" " https_proxy=" " python3.10 -m pip install -U pip > /dev/null
5757 [ $? -ne 0 ] && LOG " [FATAL] Update pip failed!" && exit -1
5858 # install paddle
59- pip install astor
59+ python3.10 -m pip install astor
6060 LOG " [INFO] Install paddlepaddle-develop ..."
61- python -m pip install --pre paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu118/ > /dev/null
61+ python3.10 -m pip install --pre paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu118/ > /dev/null
6262 [ $? -ne 0 ] && LOG " [FATAL] Install paddlepaddle-develop failed!" && exit -1
63- python -c " import paddle; print('[PaddlePaddle Commit]', paddle.version.commit)"
63+ python3.10 -c " import paddle; print('[PaddlePaddle Commit]', paddle.version.commit)"
6464 else
65- python ${GRAPH_NET_EXTRACT_WORKSPACE} /tools/count_sample.py
65+ python3.10 ${GRAPH_NET_EXTRACT_WORKSPACE} /tools/count_sample.py
6666 LOG " [INFO] This pull request doesn't change any paddle samples, skip the CI."
6767 fi
6868}
@@ -89,7 +89,7 @@ function check_torch_validation() {
8989 fail_name=()
9090 for model_path in ${MODIFIED_MODEL_PATHS[@]}
9191 do
92- python -m graph_net.torch.validate --model-path ${GRAPH_NET_EXTRACT_WORKSPACE} /${model_path} >&2
92+ python3.10 -m graph_net.torch.validate --model-path ${GRAPH_NET_EXTRACT_WORKSPACE} /${model_path} >&2
9393 [ $? -ne 0 ] && fail_name[${# fail_name[@]} ]=" ${model_path} "
9494 done
9595 local failed_cnt=${# fail_name[@]}
@@ -125,7 +125,7 @@ function check_paddle_validation() {
125125 fail_name=()
126126 for model_path in ${MODIFIED_MODEL_PATHS[@]}
127127 do
128- python -m graph_net.paddle.validate --model-path ${GRAPH_NET_EXTRACT_WORKSPACE} /${model_path} >&2
128+ python3.10 -m graph_net.paddle.validate --model-path ${GRAPH_NET_EXTRACT_WORKSPACE} /${model_path} >&2
129129 [ $? -ne 0 ] && fail_name[${# fail_name[@]} ]=" ${model_path} "
130130 done
131131 local failed_cnt=${# fail_name[@]}
@@ -165,7 +165,7 @@ function main() {
165165 check_validation_info=$( check_paddle_validation)
166166 check_validation_code=$?
167167 summary_problems $check_validation_code " $check_validation_info "
168- python ${GRAPH_NET_EXTRACT_WORKSPACE} /tools/count_sample.py
168+ python3.10 ${GRAPH_NET_EXTRACT_WORKSPACE} /tools/count_sample.py
169169 LOG " [INFO] check_validation run success and no error!"
170170}
171171
0 commit comments