build.sh 709 B

123456789101112131415161718192021222324252627
  1. # 是否使用GPU(即是否使用 CUDA)
  2. WITH_GPU=ON
  3. # 是否集成 TensorRT(仅WITH_GPU=ON 有效)
  4. WITH_TENSORRT=OFF
  5. # Paddle 预测库路径
  6. PADDLE_DIR=/path/to/fluid_inference/
  7. # CUDA 的 lib 路径
  8. CUDA_LIB=/path/to/cuda/lib/
  9. # CUDNN 的 lib 路径
  10. CUDNN_LIB=/path/to/cudnn/lib/
  11. # OPENCV 路径, 如果使用自带预编译版本可不修改
  12. OPENCV_DIR=$(pwd)/deps/opencv3gcc4.8/
  13. sh $(pwd)/scripts/bootstrap.sh
  14. # 以下无需改动
  15. rm -rf build
  16. mkdir -p build
  17. cd build
  18. cmake .. \
  19. -DWITH_GPU=${WITH_GPU} \
  20. -DWITH_TENSORRT=${WITH_TENSORRT} \
  21. -DPADDLE_DIR=${PADDLE_DIR} \
  22. -DCUDA_LIB=${CUDA_LIB} \
  23. -DCUDNN_LIB=${CUDNN_LIB} \
  24. -DOPENCV_DIR=${OPENCV_DIR}
  25. make