而对于一些未显式给出的库如TensorRT,大可以自己去写一份FindTensorRT.cmake文件,然后使用find_package命令查找,正如上交所做。 变量的使用 CMakeLists.txt中可以使用变量来简化代码,提高可读性。 使用set命令可以定义并初始化变量。 set(变量名 值) 这里的值可以是字符串、数字等,可以有多个值,用空格隔开。 set...
set(LLVM_DIR yourpath/llvm-7.0/lib/cmake/llvm) find_package(LLVM CONFIG REQUIRED) 1. 2.
find_package(Boost REQUIRED COMPONENTS system filesystem thread) find_package这里有更多细节: REQUIRED:可选字段,表示一定要找到包,找不到的话立即停掉整个cmake。 COMPONENTS:可选字段,表示查找包中的必须要找到的组件,有任何一个找不到就算失败。 find_package先在CMAKE_MODULE_PATH变量对应的路径中查找,如果路...
cmake_minimum_required(VERSION 3.12) project(yolov9-tensorrt) find_package(CUDA REQUIRED) # Set C++ standard set(CMAKE_CXX_STANDARD 17) add_definitions(-DAPI_EXPORTS) # Include CUDA directories include_directories(${CUDA_INCLUDE_DIRS}) include_directories(src/) # Add source files set(SOURCES ...
set(TENSORRT_ROOT_DIR /root/trt_projects/TensorRT-8.6.1.6) if (ENABLE_CUDA) # CUDA依赖配置 include(${CMAKE_SOURCE_DIR}/cmake/cuda.cmake) endif () if (ENABLE_TENSORRT) # TensorRT依赖配置 include(${CMAKE_SOURCE_DIR}/cmake/tensorrt.cmake) endif () find_package(CUDA REQUIRED) find_p...
onnx2tensorrt:https://github.com/onnx/onnx-tensorrt 在安装onnx2tensorrt的过程中,需要使用cmake,由于我的版本比较低,因此需要升级cmake的版本 编译: cmake .. -DTENSORRT_ROOT=../tensorrt_root && make -j (yolov4) shl@zhihui-mint:~/shl_res/1_project/yolov4-deepsort/checkpoints/onnx-tensorrt...
#include "tensorrt_llm/runtime/gptSession.h" #include "tensorrt_llm/common/stringUtils.h" #include "stdio.h" using namespace tensorrt_llm::runtime; int main() { printf("include test\n"); GptSession session(); return 0; } cmakelists ...
2、包含cuda和tensorrt # CUDA find_package(CUDA REQUIRED) message(STATUS "Find CUDA include at ${CUDA_INCLUDE_DIRS}") message(STATUS "Find CUDA libraries: ${CUDA_LIBRARIES}") # TensorRT set(TENSORRT_ROOT /data/xxx/TensorRT-7.0.0.11/) find_path(TENSORRT_INCLUDE_DIR NvInfer.h HINTS ${TENSORR...
find_package(Qt5 REQUIRED COMPONENTS Core Widgets REQUIRED) set(CMAKE_AUTOMOC ON) # 自动开启MOC编译器 set(CMAKE_AUTORCC ON)# 自动开启RCC编译器 set(CMAKE AUTOUIC ON)# 自动开启UIC编译器 # 将下面的源文件改成自己的源文件 set(CPP_SOURCE main.cpp mainwin.cpp) add_executable(start ${CPP_SOU...
/usr/bin/cc USE_TENSORRT : OFF USE_ROCM : OFF USE_EIGEN_FOR_BLAS : ON USE_FBGEMM : OFF USE_FFMPEG : OFF USE_GFLAGS : OFF USE_GLOG : OFF USE_LEVELDB : OFF USE_LITE_PROTO : OFF USE_LMDB : OFF USE_METAL : OFF USE_MKL : OFF USE_MKLDNN : OFF USE_NCCL : ON USE_SYSTEM_...