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博客

相关推荐
程序员陆通42 分钟前
独立开发A/B测试实用教程
人工智能·ai编程
knowfoot44 分钟前
硬核拆解!跟着公式“走”一遍,你也能彻底看懂神经网络
人工智能·神经网络
FF-Studio1 小时前
大语言模型(LLM)课程学习(Curriculum Learning)、数据课程(data curriculum)指南:从原理到实践
人工智能·python·深度学习·神经网络·机器学习·语言模型·自然语言处理
DDDDDouble1 小时前
<二>Sping-AI alibaba 入门-记忆聊天及持久化
java·人工智能
PyAIExplorer1 小时前
图像处理中的插值方法:原理与实践
图像处理·人工智能
狗头大军之江苏分军1 小时前
疑似华为盘古AI大模型翻车造假风波【实时记录篇】
人工智能·机器学习·程序员
Mr.Winter`1 小时前
轨迹优化 | 基于激光雷达的欧氏距离场ESDF地图构建(附ROS C++仿真)
c++·人工智能·机器人·自动驾驶·ros·ros2·具身智能
机器之心2 小时前
刚刚,苹果基础模型团队负责人庞若鸣被Meta挖走!加入超级智能团队、年薪千万美元
人工智能
G.E.N.3 小时前
开源!RAG竞技场(2):标准RAG算法
大数据·人工智能·深度学习·神经网络·算法·llm·rag