build.sh 639 B

123456789101112131415161718192021222324252627
  1. # compile with cuda
  2. WITH_GPU=OFF
  3. # compile with tensorrt
  4. WITH_TENSORRT=OFF
  5. # path to paddle inference lib
  6. PADDLE_DIR=/path/to/paddle_inference_lib/
  7. # path to cuda lib
  8. CUDA_LIB=/path/to/cuda_lib/
  9. # path to cudnn lib
  10. CUDNN_LIB=/path/to/cudnn_lib/
  11. # path to opencv lib
  12. OPENCV_DIR=$(pwd)/deps/opencv3gcc4.8/
  13. sh $(pwd)/scripts/bootstrap.sh
  14. rm -rf build
  15. mkdir -p build
  16. cd build
  17. cmake .. \
  18. -DWITH_GPU=${WITH_GPU} \
  19. -DWITH_TENSORRT=${WITH_TENSORRT} \
  20. -DPADDLE_DIR=${PADDLE_DIR} \
  21. -DCUDA_LIB=${CUDA_LIB} \
  22. -DCUDNN_LIB=${CUDNN_LIB} \
  23. -DOPENCV_DIR=${OPENCV_DIR} \
  24. -DWITH_STATIC_LIB=OFF
  25. make