36 lines
799 B
Bash
Raw Normal View History

2021-08-10 07:58:01 +00:00
set -o errexit
if [ $# != 1 ] ; then
echo "USAGE: $0 MODE (one of ['det', 'rec', 'system'])"
echo " e.g.: $0 system"
exit 1;
fi
# MODE be one of ['det', 'rec', 'system']
MODE=$1
cp CMakeLists_$MODE.txt CMakeLists.txt
OPENCV_DIR=/paddle/git/new/PaddleOCR/deploy/cpp_infer/opencv-3.4.7/opencv3/
LIB_DIR=/paddle/git/new/PaddleOCR/deploy/cpp_infer/paddle_inference/
CUDA_LIB_DIR=/usr/local/cuda/lib64/
CUDNN_LIB_DIR=/usr/lib/x86_64-linux-gnu/
2020-07-12 17:21:47 +00:00
BUILD_DIR=build
rm -rf ${BUILD_DIR}
mkdir ${BUILD_DIR}
cd ${BUILD_DIR}
cmake .. \
-DPADDLE_LIB=${LIB_DIR} \
-DWITH_MKL=ON \
-DWITH_GPU=OFF \
-DWITH_STATIC_LIB=OFF \
2021-05-26 07:02:47 +00:00
-DWITH_TENSORRT=OFF \
2020-07-12 18:48:10 +00:00
-DOPENCV_DIR=${OPENCV_DIR} \
2020-07-12 17:21:47 +00:00
-DCUDNN_LIB=${CUDNN_LIB_DIR} \
-DCUDA_LIB=${CUDA_LIB_DIR} \
2021-05-26 06:49:17 +00:00
-DTENSORRT_DIR=${TENSORRT_DIR} \
2020-07-12 17:21:47 +00:00
make -j