25 lines
543 B
Bash
25 lines
543 B
Bash
|
|
||
|
|
LIB_DIR=/paddle/code/gry/Paddle/build/fluid_inference_install_dir/
|
||
|
|
CUDA_LIB_DIR=/usr/local/cuda/lib64
|
||
|
|
CUDNN_LIB_DIR=/usr/lib/x86_64-linux-gnu/
|
||
|
|
TENSORRT_ROOT_DIR=YOUR_TENSORRT_ROOT_DIR
|
||
|
|
|
||
|
|
BUILD_DIR=build
|
||
|
|
rm -rf ${BUILD_DIR}
|
||
|
|
mkdir ${BUILD_DIR}
|
||
|
|
cd ${BUILD_DIR}
|
||
|
|
cmake .. \
|
||
|
|
-DPADDLE_LIB=${LIB_DIR} \
|
||
|
|
-DWITH_MKL=ON \
|
||
|
|
-DDEMO_NAME=ocr_system \
|
||
|
|
-DWITH_GPU=OFF \
|
||
|
|
-DWITH_STATIC_LIB=OFF \
|
||
|
|
-DUSE_TENSORRT=OFF \
|
||
|
|
-DCUDNN_LIB=${CUDNN_LIB_DIR} \
|
||
|
|
-DCUDA_LIB=${CUDA_LIB_DIR} \
|
||
|
|
-DTENSORRT_ROOT=YOUR_TENSORRT_ROOT_DIR
|
||
|
|
|
||
|
|
make -j
|
||
|
|
|
||
|
|
|