win yolov5.7 tensorRT推理

news/2024/7/11 1:08:18 标签: YOLO, 人工智能

安装TensorRT

下载tensorrt8.xx版本,适用于Windows的cuda11.x的版本

  • 官方下载地址icon-default.png?t=N7T8https://developer.nvidia.com/nvidia-tensorrt-8x-download


把tensorRT里面的bin、include、lib添加到本机CUDA中,CUDA需要加入环境变量中

 配置虚拟环境

torch的版本要和CUDA一致,

CUDA是11.7,

torch:                  torch-2.0.0+cu117-cp39-cp39-win_amd64.whl

torchvision:         torchvision-0.15.0+cu117-cp39-cp39-win_amd64.whl

  找到TensorRT-8.6.1.6\python/python/tensorrt-8.6.1-cp39-none-win_amd64.whl

pip install nvidia-pyindex 
pip install tensorrt-8.6.1-cp39-none-win_amd64.whl

pip install -r requirements.txt

YOLOv5.7的export.py中修改要转换的模型和导出格式

 编译opencv:

Windows 下编译 OpenCV 和 OpenCV-contrib_windows编译opencv-CSDN博客

【VS2019+OpenCV4.5.1+OpenCV_contrib4.5.1安装+配置保姆式教程】_Jay_z在造梦的博客-CSDN博客

在debug下报错:错误    LNK1104    无法打开文件“python310_d.lib”环境中没有安装python的debug版本,换成release环境。

GitHub - Monday-Leo/Yolov5_Tensorrt_Win10: A simple implementation of tensorrt yolov5 python/c++🔥icon-default.png?t=N7T8https://github.com/Monday-Leo/Yolov5_Tensorrt_Win10

报错: No module named ‘utils‘,untils库的安装,不是pip和conda指令,特别可以的教程,还有为什么不能的原因,绝对良心。_no module named 'utils-CSDN博客


http://www.niftyadmin.cn/n/5107636.html

相关文章

使用自定义 PyTorch 运算符优化深度学习数据输入管道

在这篇文章[1]中,我们讨论 PyTorch 对创建自定义运算符的支持,并演示它如何帮助我们解决数据输入管道的性能瓶颈、加速深度学习工作负载并降低训练成本。 构建 PyTorch 扩展 PyTorch 提供了多种创建自定义操作的方法,包括使用自定义模块和/或…

Apollo:前端开发者的全栈探索之旅

前言 「作者主页」:雪碧有白泡泡 「个人网站」:雪碧的个人网站 「推荐专栏」: ★java一站式服务 ★ ★ React从入门到精通★ ★前端炫酷代码分享 ★ ★ 从0到英雄,vue成神之路★ ★ uniapp-从构建到提升★ ★ 从0到英雄&#xff…

零基础制作预约小程序,微信小程序预约服务指南

随着互联网的发展,越来越多的服务开始转移到线上。预约服务也是其中之一。通过微信小程序,商家可以提供更加便捷的预约服务,让客户随时随地预约商品或服务。本文将介绍如何零基础制作预约小程序,包括使用第三方制作平台、选择合适…

Vue中 使用 Scss 实现配置、切换主题

1. 样式文件目录介绍 本项目中的公共样式文件均位于 src/assets/css 目录下,其中 index.scss是总的样式文件的汇总入口 ,common.scss 是供全局使用的一些基本样式(常量), _theme.scss、_handle.scss 两个文件是进行主题颜色配置的…

缓存失效方案

一、背景 WRITE : 数据写入Mysql 和 Redis缓存, READ:先从 Redis 缓存中取数据,拿不到再从Mysql中加载,更新到Redis 上图第三阶段,接收Mysql的binlog变更消息,可以参考阿里的 Canal&#xff0…

【27】c++设计模式——>迭代器模式(1)

迭代器实现通常包含两个主要组件:迭代器和聚合对象,聚合对象一般是vector,list,set,map等,迭代器负责在聚合对象上进行遍历,并提供了一种统一的访问元素的方法。聚合对象用来存储,并…

如何制作一个仅Python依靠Python第三方库的听写程序

导入所需的库 import speech_recognition as sr创建一个Recognizer对象 r = sr.Recognizer()使用麦克风录音 with sr.Microphone() as source:print("Please speak your sentence:")audio = r.listen(source)使用Google Speech Recognition进行语音识别 try:sent…

FutureTask配合Thread实现处理有返回结果的源码、逻辑与架构分析

文章目录 1.介绍2.使用示例3.执行过程描述4.整体的关系5.涉及到的核心源码(只提取了关键代码)5.1 Callable5.2 RunnableFuture5.3 FutureTask5.4 Thread 1.介绍 FutureTask 能够接收 Callable 类型的参数,用来处理有返回结果的情况。 2.使用…