build.sh 530 Bytes
Newer Older
LDOUBLEV's avatar
LDOUBLEV committed
1
OPENCV_DIR=/paddle/Paddle/opencv-3.4.7/opencv3
LDOUBLEV's avatar
LDOUBLEV committed
2
LIB_DIR=your inference dir
LDOUBLEV's avatar
LDOUBLEV committed
3
4
CUDA_LIB_DIR=/usr/local/cuda/lib64
CUDNN_LIB_DIR=/usr/lib/x86_64-linux-gnu
LDOUBLEV's avatar
LDOUBLEV committed
5
TENSORRT_DIR=your trt dir
littletomatodonkey's avatar
littletomatodonkey committed
6
7
8
9
10
11
12
13
14
15

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 \
LDOUBLEV's avatar
LDOUBLEV committed
16
    -DWITH_TENSORRT=OFF \
littletomatodonkey's avatar
littletomatodonkey committed
17
    -DOPENCV_DIR=${OPENCV_DIR} \
littletomatodonkey's avatar
littletomatodonkey committed
18
19
    -DCUDNN_LIB=${CUDNN_LIB_DIR} \
    -DCUDA_LIB=${CUDA_LIB_DIR} \
LDOUBLEV's avatar
LDOUBLEV committed
20
    -DTENSORRT_DIR=${TENSORRT_DIR} \
littletomatodonkey's avatar
littletomatodonkey committed
21
22

make -j