build.sh 1.0 KB
Newer Older
C
channings 已提交
1 2 3 4 5
# 是否使用GPU(即是否使用 CUDA)
WITH_GPU=OFF
# 使用MKL or openblas
WITH_MKL=ON
# 是否集成 TensorRT(仅WITH_GPU=ON 有效)
6
WITH_TENSORRT=OFF
C
channings 已提交
7 8 9 10 11 12 13 14 15 16 17
# TensorRT 的lib路径
TENSORRT_DIR=/path/to/TensorRT/
# Paddle 预测库路径
PADDLE_DIR=/path/to/fluid_inference/
# Paddle 的预测库是否使用静态库来编译
# 使用TensorRT时,Paddle的预测库通常为动态库
WITH_STATIC_LIB=OFF
# CUDA 的 lib 路径
CUDA_LIB=/path/to/cuda/lib/
# CUDNN 的 lib 路径
CUDNN_LIB=/path/to/cudnn/lib/
18

C
channings 已提交
19 20 21
# OPENCV 路径, 如果使用自带预编译版本可不修改
sh $(pwd)/scripts/bootstrap.sh  # 下载预编译版本的opencv
OPENCV_DIR=$(pwd)/deps/opencv3gcc4.8/
22

C
channings 已提交
23
# 以下无需改动
24 25 26 27
rm -rf build
mkdir -p build
cd build
cmake .. \
C
channings 已提交
28 29 30 31
    -DWITH_GPU=${WITH_GPU} \
    -DWITH_MKL=${WITH_MKL} \
    -DWITH_TENSORRT=${WITH_TENSORRT} \
    -DTENSORRT_DIR=${TENSORRT_DIR} \
32
    -DPADDLE_DIR=${PADDLE_DIR} \
C
channings 已提交
33
    -DWITH_STATIC_LIB=${WITH_STATIC_LIB} \
34
    -DCUDA_LIB=${CUDA_LIB} \
C
channings 已提交
35 36
    -DCUDNN_LIB=${CUDNN_LIB} \
    -DOPENCV_DIR=${OPENCV_DIR}
37
make