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

相关推荐
网络工程小王4 分钟前
【Python数据分析基础】
大数据·数据库·人工智能·学习
skilllite作者6 分钟前
开源项目推荐SkillLite,项目取得阶段性成果总结
人工智能
二十雨辰14 分钟前
[RAG]-智能体开发
人工智能·ai
第七页独白20 分钟前
全星研发项目管理 APQP 软件系统:驱动高端制造研发数智化升级
人工智能
FluxMelodySun21 分钟前
机器学习(二十七) 降维:度量学习与随机梯度下降法求解
人工智能·学习·机器学习
蒸汽求职24 分钟前
【蒸汽教育求职干货】OPT只剩3个月还没找到工作,怎么办?——留学生IT求职的“紧急预案”
人工智能·经验分享·面试·职场和发展·美国求职
陆业聪24 分钟前
从推理到行动:Agent 范式的真正跃迁正在发生
人工智能·aigc
蒸汽求职27 分钟前
【蒸汽教育求职分享】美国IT面试的Behavioral Question:STAR法则人人都知道,但90%的人用错了
人工智能·面试·职场和发展·github·求职招聘·留学生求职
人工智能研究所32 分钟前
字节开源 DeerFlow 2.0——登顶 GitHub Trending 1,让 AI 可做任何事情
人工智能·深度学习·开源·github·ai agent·字节跳动·deerflow2.0
江瀚视野32 分钟前
美丽田园经调净利大增41%,全方位增长未来何在?
大数据·人工智能