jetpack5.02版本自带默认cuda cudnn，tensorrt路径

Godlovesea

已于 2023-05-30 18:14:27 修改

阅读量1k

点赞数 1

文章标签： opencv c++ 人工智能

于 2023-05-30 15:47:35 首次发布

本文链接：https://blog.csdn.net/weixin_42064949/article/details/130950183

版权

tensorrt源文件路径为/usr/src/tensorrt/bin 下trtexec 可进行onnx转换engine 命令如下：

sudo ./trtexec --onnx=yolov7-tiny.onnx --saveEngine=./yolov7-tiny-fp16.engine --fp16 如果转换fp32 就不加--fp16参数，好像--fp32会报错

include_directories(${PROJECT_SOURCE_DIR}/include)
# include and link dirs of cuda and tensorrt, you need adapt them if yours are different
# cuda
include_directories(/usr/local/cuda-11.4/include)
link_directories(/usr/local/cuda-11.4/lib64)
# tensorrt
include_directories(/usr/include/x86_64-linux-gnu/)
link_directories(/usr/lib/x86_64-linux-gnu/)

set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -Wall -Ofast -Wfatal-errors -D_MWAITXINTRIN_H_INCLUDED")

find_package(OpenCV)
include_directories(${OpenCV_INCLUDE_DIRS})

add_executable(yolo ${PROJECT_SOURCE_DIR}/main.cpp yolo.hpp)
target_link_libraries(yolo nvinfer)
target_link_libraries(yolo cudart)
target_link_libraries(yolo ${OpenCV_LIBS})