tensorrtx onnx的配置使用

1、windows缺少dirent.h（遍历文件和目录的API，通常unix系统会提供），先下载dirent.h放到

C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.29.30133\include 里。

下载地址 dirent/dirent.h at master · tronkko/dirent (github.com)

2、下载tensorrt、cuda、cudnn（解压放到相应的cuda目录），opencv（最新版本即可），然后dll文件的目录添加到系统环境变量path中

通过tensorrt包名就知道对应的其他版本，如 TensorRT-8.4.1.5.Windows10.x86_64.cuda-11.6.cudnn8.4

3、tensorrtx案例源码 wang-xinyu/tensorrtx: Implementation of popular deep learning networks with TensorRT network definition API (github.com)

VS打开CMakeLists.txt，运行。

可能的报错及解决：

VC++：C2589 “(”:“::”右边的非法标记问题_smartisong的博客-CSDN博客

c/c++：cuda编译时 error: : identifier “xxx“ is undefined 的错误_wjjun_89的博客-CSDN博客

yolov7的CMakeLists.txt

cmake_minimum_required(VERSION 3.10)

project(yolov7)

add_definitions(-std=c++11)
add_definitions(-DAPI_EXPORTS)
add_definitions(-DNOMINMAX) #解决msvc宏定义冲突造成的编译错误
set(CMAKE_CXX_STANDARD 11)
set(CMAKE_BUILD_TYPE Debug)

set(CMAKE_CUDA_COMPILER "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.8/bin/nvcc.exe") # 路径中有空格，必须添加双引号
enable_language(CUDA)

include_directories(${PROJECT_SOURCE_DIR}/include)
include_directories(${PROJECT_SOURCE_DIR}/plugin)

# include and link dirs of cuda and tensorrt, you need adapt them if yours are different
message("embed_platform off")
# cuda
include_directories("C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.8/include")
link_directories("C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.8/lib/x64")

# tensorrt
include_directories("C:/TensorRT/include")
link_directories("C:/TensorRT/lib")

add_library(myplugins SHARED ${PROJECT_SOURCE_DIR}/plugin/yololayer.cu)
target_link_libraries(myplugins nvinfer cudart)

# opencv
set(OpenCV_DIR "C:/opencv/build")    #该地址为OpenCVConfig.cmake所在的目录地址
find_package(OpenCV)
include_directories(${OpenCV_INCLUDE_DIRS})

file(GLOB_RECURSE SRCS ${PROJECT_SOURCE_DIR}/src/*.cpp ${PROJECT_SOURCE_DIR}/src/*.cu)
add_executable(yolov7 main.cpp ${SRCS})

target_link_libraries(yolov7 nvinfer)
target_link_libraries(yolov7 cudart)
target_link_libraries(yolov7 myplugins)
target_link_libraries(yolov7 ${OpenCV_LIBS})