File tree Expand file tree Collapse file tree
IdentityConvIPluginV2IOExt Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -8,6 +8,11 @@ set(CMAKE_CXX_STANDARD_REQUIRED ON)
88# Find CUDA Toolkit
99find_package (CUDAToolkit REQUIRED )
1010
11+ # TensorRT NvInfer related libs
12+ find_library (NVINFER_LIB nvinfer HINTS /usr/lib/x86_64-linux-gnu/ PATH_SUFFIXES lib lib64 REQUIRED )
13+ find_library (NVINFER_PLUGIN_LIB nvinfer_plugin HINTS /usr/lib/x86_64-linux-gnu/ PATH_SUFFIXES lib lib64 REQUIRED )
14+ find_library (NVONNXPARSER_LIB nvonnxparser HINTS /usr/lib/x86_64-linux-gnu/ PATH_SUFFIXES lib lib64 REQUIRED )
15+
1116add_executable (build_engine build_engine.cpp )
1217target_link_libraries (build_engine PRIVATE ${NVINFER_LIB} ${NVONNXPARSER_LIB} ${NVINFER_PLUGIN_LIB} CUDA::cudart )
1318
Original file line number Diff line number Diff line change @@ -11,7 +11,6 @@ find_package(CUDAToolkit REQUIRED)
1111# TensorRT NvInfer related libs
1212find_library (NVINFER_LIB nvinfer HINTS /usr/lib/x86_64-linux-gnu/ PATH_SUFFIXES lib lib64 REQUIRED )
1313find_library (NVINFER_PLUGIN_LIB nvinfer_plugin HINTS /usr/lib/x86_64-linux-gnu/ PATH_SUFFIXES lib lib64 REQUIRED )
14- find_library (NVONNXPARSER_LIB nvonnxparser HINTS /usr/lib/x86_64-linux-gnu/ PATH_SUFFIXES lib lib64 REQUIRED )
1514
1615# Add all the source files in the current directory to build the library
1716add_library (
Original file line number Diff line number Diff line change @@ -11,7 +11,6 @@ find_package(CUDAToolkit REQUIRED)
1111# TensorRT NvInfer related libs
1212find_library (NVINFER_LIB nvinfer HINTS /usr/lib/x86_64-linux-gnu/ PATH_SUFFIXES lib lib64 REQUIRED )
1313find_library (NVINFER_PLUGIN_LIB nvinfer_plugin HINTS /usr/lib/x86_64-linux-gnu/ PATH_SUFFIXES lib lib64 REQUIRED )
14- find_library (NVONNXPARSER_LIB nvonnxparser HINTS /usr/lib/x86_64-linux-gnu/ PATH_SUFFIXES lib lib64 REQUIRED )
1514
1615# Add all the source files in the current directory to build the library
1716add_library (
You can’t perform that action at this time.
0 commit comments