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})

 yolov7.exe -s yolov7-tiny.weights yolov7-tiny.engine t

yolov7的 TensorRT c++推理,win10, cuda11.4.3 ,cudnn8.2.4.15,tensorrt8.2.1.8。_vokxchh的博客-CSDN博客

 

posted @ 2023-03-21 16:35  夕西行  阅读(123)  评论(0编辑  收藏  举报