build.sh 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. # 是否使用GPU(即是否使用 CUDA)
  2. WITH_GPU=OFF
  3. # 使用MKL or openblas
  4. WITH_MKL=ON
  5. # 是否集成 TensorRT(仅WITH_GPU=ON 有效)
  6. WITH_TENSORRT=OFF
  7. # TensorRT 的lib路径
  8. TENSORRT_DIR=/path/to/TensorRT/
  9. # Paddle 预测库路径
  10. PADDLE_DIR=/path/to/fluid_inference/
  11. # Paddle 的预测库是否使用静态库来编译
  12. # 使用TensorRT时,Paddle的预测库通常为动态库
  13. WITH_STATIC_LIB=OFF
  14. # CUDA 的 lib 路径
  15. CUDA_LIB=/path/to/cuda/lib/
  16. # CUDNN 的 lib 路径
  17. CUDNN_LIB=/path/to/cudnn/lib/
  18. # OPENCV 路径, 如果使用自带预编译版本可不修改
  19. OPENCV_DIR=$(pwd)/deps/opencv3gcc4.8/
  20. sh $(pwd)/scripts/bootstrap.sh
  21. # 以下无需改动
  22. rm -rf build
  23. mkdir -p build
  24. cd build
  25. cmake .. \
  26. -DWITH_GPU=${WITH_GPU} \
  27. -DWITH_MKL=${WITH_MKL} \
  28. -DWITH_TENSORRT=${WITH_TENSORRT} \
  29. -DTENSORRT_DIR=${TENSORRT_DIR} \
  30. -DPADDLE_DIR=${PADDLE_DIR} \
  31. -DWITH_STATIC_LIB=${WITH_STATIC_LIB} \
  32. -DCUDA_LIB=${CUDA_LIB} \
  33. -DCUDNN_LIB=${CUDNN_LIB} \
  34. -DOPENCV_DIR=${OPENCV_DIR}
  35. make