2021-11-15 12:19:21 +08:00
|
|
|
OPENCV_DIR=${opencv_install_dir}
|
|
|
|
|
LIB_DIR=${paddle_inference_dir}
|
2021-11-02 19:49:55 +08:00
|
|
|
CUDA_LIB_DIR=/usr/local/cuda/lib64
|
|
|
|
|
CUDNN_LIB_DIR=/usr/lib/x86_64-linux-gnu/
|
2021-11-15 12:19:21 +08:00
|
|
|
FAISS_DIR=${faiss_install_dir}
|
2021-11-09 14:25:01 +08:00
|
|
|
FAISS_WITH_MKL=OFF
|
2021-11-02 19:49:55 +08: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 \
|
|
|
|
|
-DUSE_TENSORRT=OFF \
|
|
|
|
|
-DOPENCV_DIR=${OPENCV_DIR} \
|
|
|
|
|
-DCUDNN_LIB=${CUDNN_LIB_DIR} \
|
|
|
|
|
-DCUDA_LIB=${CUDA_LIB_DIR} \
|
2021-11-08 23:26:13 +08:00
|
|
|
-DFAISS_DIR=${FAISS_DIR} \
|
2021-11-09 14:25:01 +08:00
|
|
|
-DFAISS_WITH_MKL=${FAISS_WITH_MKL}
|
2021-11-02 19:49:55 +08:00
|
|
|
|
2021-12-07 14:56:12 +08:00
|
|
|
make -j
|
|
|
|
|
cd ..
|
