22 lines
480 B
Bash
Raw Normal View History

2021-05-26 18:49:44 +08:00
OPENCV_DIR=your_opencv_dir
LIB_DIR=your_paddle_inference_dir
CUDA_LIB_DIR=your_cuda_lib_dir
CUDNN_LIB_DIR=your_cudnn_lib_dir
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