build.sh 639 字节
Newer Older
L
LDOUBLEV 已提交
1 2 3 4 5 6
OPENCV_DIR=/paddle/Paddle/opencv-3.4.7/opencv3
LIB_DIR=/paddle/OCR/debug/paddle_inference
#LIB_DIR=/paddle/Paddle/inference/2.0.2/paddle_inference
CUDA_LIB_DIR=/usr/local/cuda/lib64
CUDNN_LIB_DIR=/usr/lib/x86_64-linux-gnu
TENSORRT_DIR=/paddle/Paddle/package/TensorRT/TensorRT-6.0.1.5/
littletomatodonkey's avatar
littletomatodonkey 已提交
7 8 9 10 11 12 13 14 15 16

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 \
L
LDOUBLEV 已提交
17
    -DWITH_TENSORRT=ON \
littletomatodonkey's avatar
littletomatodonkey 已提交
18
    -DOPENCV_DIR=${OPENCV_DIR} \
littletomatodonkey's avatar
littletomatodonkey 已提交
19 20
    -DCUDNN_LIB=${CUDNN_LIB_DIR} \
    -DCUDA_LIB=${CUDA_LIB_DIR} \
L
LDOUBLEV 已提交
21
    -DTENSORRT_DIR=${TENSORRT_DIR} \
littletomatodonkey's avatar
littletomatodonkey 已提交
22 23

make -j