win yolov5.7 tensorRT推理

安装TensorRT

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

把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

bash 复制代码
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++🔥https://github.com/Monday-Leo/Yolov5_Tensorrt_Win10

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

相关推荐
aneasystone本尊几秒前
学习大模型训练的监督微调:SFT 与 LoRA
人工智能
lucas_AI2 分钟前
Mira Murati 公司第二代模型来了:Inkling 2 Small 首日闭门、下周开源,AA Index 直接冲到 31
人工智能
byte轻骑兵2 分钟前
【LMP】规范精讲[13]: 参数的户口本、公钥的集装箱与出厂的兜底值
人工智能·网络协议·人机交互·lmp·蓝牙链路管理
天天被压力3 分钟前
【A股分钟级高频数据完全手册 #03】股票历史1分钟K线数据接口怎么用:Pro版批量拉取与本地缓存
java·人工智能·python
lucas_AI9 分钟前
INT4 提速 2 倍的真相:batch 一大就「穿帮」,9B–20B 模型量化部署实测盘点
人工智能
深蓝AI3 小时前
Gemini 4 Carbon 登顶却买不到:59 分拿下智能指数第一,单任务 7.76 美元,幻觉率从 4% 涨到 11%
人工智能·程序员
云-缩影9 小时前
文生图技术落地:从广告到游戏,哪些场景真的能用
人工智能·游戏
冬奇Lab12 小时前
一天一个开源项目(第254篇):NVIDIA Auto Ontology —— 给数据 Agent 搭一层「可认证的语义层」,答不了就老实说答不了
人工智能
其实防守也摸鱼12 小时前
个人学习 Pwn 思路:从基础到漏洞利用的进阶路径
大数据·前端·人工智能·chrome·算法·安全·网络安全