| 123456789101112131415161718192021222324252627 |
- # 是否使用GPU(即是否使用 CUDA)
- WITH_GPU=ON
- # 是否集成 TensorRT(仅WITH_GPU=ON 有效)
- WITH_TENSORRT=OFF
- # Paddle 预测库路径
- PADDLE_DIR=/path/to/fluid_inference/
- # CUDA 的 lib 路径
- CUDA_LIB=/path/to/cuda/lib/
- # CUDNN 的 lib 路径
- CUDNN_LIB=/path/to/cudnn/lib/
- # OPENCV 路径, 如果使用自带预编译版本可不修改
- OPENCV_DIR=$(pwd)/deps/opencv3gcc4.8/
- sh $(pwd)/scripts/bootstrap.sh
- # 以下无需改动
- rm -rf build
- mkdir -p build
- cd build
- cmake .. \
- -DWITH_GPU=${WITH_GPU} \
- -DWITH_TENSORRT=${WITH_TENSORRT} \
- -DPADDLE_DIR=${PADDLE_DIR} \
- -DCUDA_LIB=${CUDA_LIB} \
- -DCUDNN_LIB=${CUDNN_LIB} \
- -DOPENCV_DIR=${OPENCV_DIR}
- make
|