build.sh 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. # 是否使用GPU(即是否使用 CUDA)
  2. WITH_GPU=ON
  3. # 使用MKL or openblas
  4. WITH_MKL=ON
  5. # 是否集成 TensorRT(仅WITH_GPU=ON 有效)
  6. WITH_PADDLE_TENSORRT=OFF
  7. # TensorRT 的路径,如果需要集成TensorRT,需修改为您实际安装的TensorRT路径
  8. TENSORRT_DIR=$(pwd)/TensorRT/
  9. # Paddle 预测库路径, 请修改为您实际安装的预测库路径
  10. PADDLE_DIR=$(pwd)/paddle_inference
  11. # Paddle 的预测库是否使用静态库来编译
  12. # 使用TensorRT时,Paddle的预测库通常为动态库
  13. WITH_STATIC_LIB=OFF
  14. # CUDA 的 lib 路径
  15. CUDA_LIB=/usr/local/cuda/lib64
  16. # CUDNN 的 lib 路径
  17. CUDNN_LIB=/usr/lib/x86_64-linux-gnu
  18. # 是否加密
  19. WITH_ENCRYPTION=OFF
  20. # OPENSSL 路径
  21. OPENSSL_DIR=$(pwd)/deps/openssl-1.1.0k
  22. {
  23. bash $(pwd)/scripts/bootstrap.sh # 下载预编译版本的加密工具和opencv依赖库
  24. } || {
  25. echo "Fail to execute script/bootstrap.sh"
  26. exit -1
  27. }
  28. # OPENCV 路径, 如果使用自带预编译版本可不修改
  29. OPENCV_DIR=$(pwd)/deps/opencv3.4.6gcc4.8ffmpeg/
  30. # 以下无需改动
  31. rm -rf build
  32. mkdir -p build
  33. cd build
  34. cmake .. \
  35. -DWITH_GPU=${WITH_GPU} \
  36. -DWITH_MKL=${WITH_MKL} \
  37. -DWITH_PADDLE_TENSORRT=${WITH_PADDLE_TENSORRT} \
  38. -DTENSORRT_DIR=${TENSORRT_DIR} \
  39. -DPADDLE_DIR=${PADDLE_DIR} \
  40. -DWITH_STATIC_LIB=${WITH_STATIC_LIB} \
  41. -DCUDA_LIB=${CUDA_LIB} \
  42. -DCUDNN_LIB=${CUDNN_LIB} \
  43. -DOPENCV_DIR=${OPENCV_DIR} \
  44. -DWITH_ENCRYPTION=${WITH_ENCRYPTION} \
  45. -DOPENSSL_DIR=${OPENSSL_DIR}
  46. make -j16