diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index db8f3925879fbf8781af59ef6ec0d17b3b699d81..69e71a9e91193a6610394f021475dd3bd399804c 100644
--- a/test_tipc/prepare.sh
+++ b/test_tipc/prepare.sh
@@ -43,8 +43,7 @@ function func_get_url_file_name() {
 model_name=$(func_parser_value "${lines[1]}")
 
 # install paddleclas whl
-python_name=$(func_parser_value "${lines[2]}")
-${python_name} setup.py install
+python setup.py install
 
 if [[ ${MODE} = "cpp_infer" ]]; then
     if [ -d "./deploy/cpp/opencv-3.4.7/opencv3/" ] && [ $(md5sum ./deploy/cpp/opencv-3.4.7.tar.gz | awk -F ' ' '{print $1}') = "faa2b5950f8bee3f03118e600c74746a" ]; then
diff --git a/test_tipc/test_train_inference_python_npu.sh b/test_tipc/test_train_inference_python_npu.sh
index d1aa8ec968523ffdfe57566e57a77714eb8be9bd..6f2870ad4f02098f71aeeec740de2ba7f5a0cf89 100644
--- a/test_tipc/test_train_inference_python_npu.sh
+++ b/test_tipc/test_train_inference_python_npu.sh
@@ -24,6 +24,8 @@ sed -i "s/Global.use_gpu/Global.use_npu/g" $FILENAME
 sed -i "s/Global.use_tensorrt:True|False/Global.use_tensorrt:False/g" $FILENAME
 sed -i "s/Global.save_interval=2/Global.save_interval=1/g" $FILENAME
 sed -i "s/-o Global.epochs:lite_train_lite_infer=2/-o Global.epochs:lite_train_lite_infer=1/g" $FILENAME
+# python has been updated to version 3.9 for npu backend
+sed -i "s/python3.7/python3.9/g" $FILENAME
 
 modelname=$(echo $FILENAME | cut -d '/' -f4)
 if  [ $modelname == "PVTV2" ] || [ $modelname == "Twins" ] || [ $modelname == "SwinTransformer" ]; then