diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index 1cc03bebbf..e870501dbb 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 6007119784..bcaf8e28e0 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