build.sh 577 字节
Newer Older
D
dongshuilong 已提交
1 2
OPENCV_DIR=${opencv_install_dir}
LIB_DIR=${paddle_inference_dir}
3 4
CUDA_LIB_DIR=/usr/local/cuda/lib64
CUDNN_LIB_DIR=/usr/lib/x86_64-linux-gnu/
D
dongshuilong 已提交
5
FAISS_DIR=${faiss_install_dir}
D
dongshuilong 已提交
6
FAISS_WITH_MKL=OFF
7 8 9 10 11 12 13 14 15 16 17 18 19 20

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 \
    -DOPENCV_DIR=${OPENCV_DIR} \
    -DCUDNN_LIB=${CUDNN_LIB_DIR} \
    -DCUDA_LIB=${CUDA_LIB_DIR} \
D
dongshuilong 已提交
21
    -DFAISS_DIR=${FAISS_DIR} \
D
dongshuilong 已提交
22
    -DFAISS_WITH_MKL=${FAISS_WITH_MKL}
23

D
dongshuilong 已提交
24
make -j