| 123456789101112131415161718192021222324252627282930313233343536373839404142 |
- # 是否使用GPU(即是否使用 CUDA)
- WITH_GPU=ON
- # 使用MKL or openblas
- WITH_MKL=ON
- # 是否集成 TensorRT(仅WITH_GPU=ON 有效)
- WITH_TENSORRT=OFF
- # TensorRT 的路径,如果需要集成TensorRT,需修改为您实际安装的TensorRT路径
- TENSORRT_DIR=$(pwd)/TensorRT/
- # Paddle 预测库路径, 请修改为您实际安装的预测库路径
- PADDLE_DIR=$(pwd)/paddle_inference
- # Paddle 的预测库是否使用静态库来编译
- # 使用TensorRT时,Paddle的预测库通常为动态库
- WITH_STATIC_LIB=ON
- # CUDA 的 lib 路径
- CUDA_LIB=/usr/local/cuda/lib64
- # CUDNN 的 lib 路径
- CUDNN_LIB=/usr/lib/x86_64-linux-gnu
- {
- bash $(pwd)/scripts/bootstrap.sh # 下载预编译版本的加密工具和opencv依赖库
- } || {
- echo "Fail to execute script/bootstrap.sh"
- exit -1
- }
- # OPENCV 路径, 如果使用自带预编译版本可不修改
- OPENCV_DIR=$(pwd)/deps/opencv3.4.6gcc4.8ffmpeg/
- # 以下无需改动
- rm -rf build
- mkdir -p build
- cd build
- cmake ../tools/build_lib \
- -DWITH_GPU=${WITH_GPU} \
- -DWITH_MKL=${WITH_MKL} \
- -DWITH_TENSORRT=${WITH_TENSORRT} \
- -DTENSORRT_DIR=${TENSORRT_DIR} \
- -DPADDLE_DIR=${PADDLE_DIR} \
- -DWITH_STATIC_LIB=${WITH_STATIC_LIB} \
- -DCUDA_LIB=${CUDA_LIB} \
- -DCUDNN_LIB=${CUDNN_LIB} \
- -DOPENCV_DIR=${OPENCV_DIR}
- make install -j16
|