Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Kceshi #1850

Open
wants to merge 8 commits into
base: develop
Choose a base branch
from
Original file line number Diff line number Diff line change
Expand Up @@ -14,10 +14,14 @@ WITH_MKL=ON
WITH_GPU=ON
USE_TENSORRT=ON

LIB_DIR=/root/auto_compress/Paddle/build/paddle_inference_install_dir/
# Paddle Inference预测库路径
LIB_DIR=/work/Paddle/build/paddle_inference_install_dir
# CUDNN路径
CUDNN_LIB=/usr/lib/x86_64-linux-gnu/
# CUDA路径
CUDA_LIB=/usr/local/cuda/lib64
TENSORRT_ROOT=/root/auto_compress/trt/trt8.4/
# TensorRT安装包路径,为TRT资源包解压后的绝对路径,其中包含lib和include文件夹
TENSORRT_ROOT=/work/TensorRT-8.6.1.6

WITH_ROCM=OFF
ROCM_LIB=/opt/rocm/lib
Expand All @@ -34,4 +38,4 @@ cmake .. -DPADDLE_LIB=${LIB_DIR} \
-DCUDA_LIB=${CUDA_LIB} \
-DTENSORRT_ROOT=${TENSORRT_ROOT}

make -j
make -j