From 4c9ad75924f04900054e358d2a958688e7f334a6 Mon Sep 17 00:00:00 2001
From: Double_V <liuvv0203@163.com>
Date: Wed, 18 May 2022 16:07:58 +0800
Subject: [PATCH] [Tipc] fix prepare.sh for cpp infer (#5520)

* add pact about

* fix doc

* fix comments

* add ptq aware

* fix

* add benchmark.png

* fix png and ptq

* fix tipc

* fix train_infer_python

* delete mkldnn trt

* fix cpp

* fix comment

* fix comment

* fix

* status_check

* fix comment

* add build cpp to prepare.sh

* add  for prepare.sh

* add pip install -r requirements

* fix comment
---
 tutorials/mobilenetv3_prod/Step6/test_tipc/prepare.sh | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/tutorials/mobilenetv3_prod/Step6/test_tipc/prepare.sh b/tutorials/mobilenetv3_prod/Step6/test_tipc/prepare.sh
index 4ea5a08f35..894ff28417 100644
--- a/tutorials/mobilenetv3_prod/Step6/test_tipc/prepare.sh
+++ b/tutorials/mobilenetv3_prod/Step6/test_tipc/prepare.sh
@@ -18,7 +18,7 @@ model_name=$(func_parser_value "${lines[1]}")
 
 trainer_list=$(func_parser_value "${lines[12]}")
 
-
+pip install -r requirements.txt
 if [ ${MODE} = "lite_train_lite_infer" ];then
     # prepare lite data
     tar -xf ./test_images/lite_data.tar
@@ -55,12 +55,17 @@ elif [ ${MODE} = "serving_infer" ];then
         cd ./inference && tar xf mobilenet_v3_small_infer.tar && cd ../
     fi
 elif [ ${MODE} = "cpp_infer" ];then
+    PADDLEInfer=$3
     # wget model
     wget -nc -P  ./deploy/inference_cpp/ https://paddle-model-ecology.bj.bcebos.com/model/mobilenetv3_reprod/mobilenet_v3_small_infer.tar  --no-check-certificate
     cd ./deploy/inference_cpp/ && tar xf mobilenet_v3_small_infer.tar
-    wget -nc https://paddle-inference-lib.bj.bcebos.com/2.2.2/cxx_c/Linux/GPU/x86-64_gcc8.2_avx_mkl_cuda11.1_cudnn8.1.1_trt7.2.3.4/paddle_inference.tgz
+    if [ "" = "$PADDLEInfer" ];then
+        wget -nc https://paddle-inference-lib.bj.bcebos.com/2.2.2/cxx_c/Linux/GPU/x86-64_gcc8.2_avx_mkl_cuda11.1_cudnn8.1.1_trt7.2.3.4/paddle_inference.tgz --no-check-certificate
+    else
+        wget -nc $PADDLEInfer --no-check-certificate
+    fi
     tar zxf paddle_inference.tgz
-    wget -nc https://paddleocr.bj.bcebos.com/libs/opencv/opencv-3.4.7.tar.gz
+    wget -nc https://paddleocr.bj.bcebos.com/libs/opencv/opencv-3.4.7.tar.gz --no-check-certificate
     tar zxf opencv-3.4.7.tar.gz
     # build opencv
     cd opencv-3.4.7/