build.sh 442 Bytes
Newer Older
littletomatodonkey's avatar
littletomatodonkey committed
1
2
3
OPENCV_DIR=your_opencv_dir
LIB_DIR=your_paddle_inference_dir
CUDA_LIB_DIR=your_cuda_lib_dir
4
CUDNN_LIB_DIR=your_cudnn_lib_dir
littletomatodonkey's avatar
littletomatodonkey committed
5
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 \
    -DUSE_TENSORRT=OFF \
littletomatodonkey's avatar
littletomatodonkey committed
16
    -DOPENCV_DIR=${OPENCV_DIR} \
littletomatodonkey's avatar
littletomatodonkey committed
17
18
19
20
    -DCUDNN_LIB=${CUDNN_LIB_DIR} \
    -DCUDA_LIB=${CUDA_LIB_DIR} \

make -j