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

相关推荐
公链开发14 小时前
2026 Web3机构级风口:RWA Tokenization + ZK隐私系统定制开发全解析
人工智能·web3·区块链
wyw000014 小时前
目标检测之YOLO
人工智能·yolo·目标检测
发哥来了14 小时前
AI视频生成企业级方案选型指南:2025年核心能力与成本维度深度对比
大数据·人工智能
_codemonster14 小时前
强化学习入门到实战系列(四)马尔科夫决策过程
人工智能
北邮刘老师14 小时前
智能体治理:人工智能时代信息化系统的全新挑战与课题
大数据·人工智能·算法·机器学习·智能体互联网
laplace012314 小时前
第七章 构建自己的agent智能体框架
网络·人工智能·microsoft·agent
诗词在线14 小时前
中国古代诗词名句按主题分类有哪些?(爱国 / 思乡 / 送别)
人工智能·python·分类·数据挖掘
高锰酸钾_14 小时前
机器学习-L1正则化和L2正则化解决过拟合问题
人工智能·python·机器学习
${王小剑}14 小时前
深度学习损失函数
人工智能·深度学习
啊巴矲14 小时前
小白从零开始勇闯人工智能:机器学习初级篇(PCA数据降维)
人工智能·机器学习