当前位置: 首页 > news >正文

linux c++ 开发 tensorrt 安装

tensorrt 官方下载地址(需要注册账号登录):Log in | NVIDIA Developer

根据系统发行版和CUDA版本 (nvcc -V) 选择合适的安装包

EA(early access)版本代表抢先体验。
GA(general availability)代表稳定版,经过全面测试。
建议选用 TensorRT 最新版本的 GA release 

为了不影响系统环境,只进行单个 C++ 项目的开发,这里选用 TAR Package,解压即用,无需安装。 

tar -zxvf TensorRT-8.6.1.6.Linux.x86_64-gnu.cuda-11.8.tar.gz

cmake 文件参考

cmake_minimum_required(VERSION 3.1)set(CMAKE_CUDA_ARCHITECTURES 60 61 62 70 72 75 86)
set(CMAKE_CUDA_COMPILER /usr/local/cuda/bin/nvcc)project(yolov8 LANGUAGES CXX CUDA)set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++14 -O3")
set(CMAKE_CXX_STANDARD 14)
set(CMAKE_BUILD_TYPE Release)
option(CUDA_USE_STATIC_CUDA_RUNTIME OFF)# CUDA
find_package(CUDA REQUIRED)
message(STATUS "CUDA Libs: \n${CUDA_LIBRARIES}\n")
get_filename_component(CUDA_LIB_DIR ${CUDA_LIBRARIES} DIRECTORY)
message(STATUS "CUDA Headers: \n${CUDA_INCLUDE_DIRS}\n")# OpenCV
set(OpenCV_DIR "/home/c++/lib/opencv-4.8.1/build")
find_package(OpenCV REQUIRED)
message(STATUS "OpenCV Libs: \n${OpenCV_LIBS}\n")
message(STATUS "OpenCV Libraries: \n${OpenCV_LIBRARIES}\n")
message(STATUS "OpenCV Headers: \n${OpenCV_INCLUDE_DIRS}\n")# TensorRT
set(TensorRT_INCLUDE_DIRS "/home/c++/lib/TensorRT-8.6.1.6/include")
set(TensorRT_LIBRARIES "/home/c++/lib/TensorRT-8.6.1.6/lib")
set(TensorRT_LIB1 "/home/c++/lib/TensorRT-8.6.1.6/lib/libnvinfer.so")
set(TensorRT_LIB2 "/home/c++/lib/TensorRT-8.6.1.6/lib/libnvinfer_plugin.so")message(STATUS "TensorRT Libs: \n${TensorRT_LIBRARIES}\n")
message(STATUS "TensorRT Headers: \n${TensorRT_INCLUDE_DIRS}\n")list(APPEND INCLUDE_DIRS${CUDA_INCLUDE_DIRS}${OpenCV_INCLUDE_DIRS}${TensorRT_INCLUDE_DIRS}include)list(APPEND ALL_LIBS${CUDA_LIBRARIES}${CUDA_LIB_DIR}${OpenCV_LIBRARIES}${TensorRT_LIBRARIES})include_directories(${INCLUDE_DIRS})add_executable(${PROJECT_NAME}main.cppinclude/yolov8.hppinclude/common.hpp)link_directories(${ALL_LIBS})
target_link_libraries(${PROJECT_NAME} PRIVATE ${CUDA_LIBRARIES} ${OpenCV_LIBS} ${TensorRT_LIB1} ${TensorRT_LIB2})

http://www.lryc.cn/news/306442.html

相关文章:

  • Redis高并发分布锁实战
  • Kotlin基础——DSL
  • 《Docker 简易速速上手小册》第4章 Docker 容器管理(2024 最新版)
  • 【人脸朝向识别与分类预测】基于PNN神经网络
  • 【Python笔记-设计模式】组合模式
  • 51单片机学习(5)-----蜂鸣器的介绍与使用
  • -bash: /root/.ssh/authorized_keys: Read-only file system
  • 3,设备无关位图显示
  • 转前端了!!
  • RESTful API如何使用它构建 web 应用程序。
  • 现在学Oracle是49年入国军么?
  • 【回溯】组合问题||
  • 【c语言】字符函数和字符串函数(下)
  • 基于Java的艺培管理解决方案
  • Python算法题集_实现 Trie [前缀树]
  • pytorch简单新型模型测试参数
  • Unity中URP下实现水体(水面高光)
  • 26.HarmonyOS App(JAVA)列表对话框
  • 五种主流数据库:常用字符函数
  • 软考笔记--企业资源规划和实施
  • React歌词滚动效果(跟随音乐播放时间滚动)
  • java面试题之mybatis篇
  • Java的编程之旅19——使用idea对面相对象编程项目的创建
  • docker build基本命令
  • nginx高级配置详解
  • 小程序--分包加载
  • R语言【base】——writeLines()
  • 微信小程序-人脸检测
  • 微信小程序自制动态导航栏
  • 金融知识分享系列之:五日线