Skip to content

Commit

Permalink
[Tipc] fix prepare.sh for cpp infer (#5520)
Browse files Browse the repository at this point in the history
* 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
  • Loading branch information
LDOUBLEV authored May 18, 2022
1 parent 34f046e commit 4c9ad75
Showing 1 changed file with 8 additions and 3 deletions.
11 changes: 8 additions & 3 deletions tutorials/mobilenetv3_prod/Step6/test_tipc/prepare.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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/
Expand Down

0 comments on commit 4c9ad75

Please sign in to comment.