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

相关推荐
音视频牛哥19 小时前
超清≠清晰:视频系统里的分辨率陷阱与秩序真相
人工智能·机器学习·计算机视觉·音视频·大牛直播sdk·rtsp播放器rtmp播放器·smartmediakit
johnny23319 小时前
AI视频创作工具汇总:MoneyPrinterTurbo、KrillinAI、NarratoAI、ViMax
人工智能·音视频
Coovally AI模型快速验证19 小时前
当视觉语言模型接收到相互矛盾的信息时,它会相信哪个信号?
人工智能·深度学习·算法·机器学习·目标跟踪·语言模型
居7然19 小时前
Attention注意力机制:原理、实现与优化全解析
人工智能·深度学习·大模型·transformer·embedding
Scabbards_19 小时前
KGGEN: 用语言模型从纯文本中提取知识图
人工智能·语言模型·自然语言处理
LeonDL16820 小时前
【通用视觉框架】基于C#+Winform+OpencvSharp开发的视觉框架软件,全套源码,开箱即用
人工智能·c#·winform·opencvsharp·机器视觉软件框架·通用视觉框架·机器视觉框架
AI纪元故事会20 小时前
《目标检测全解析:从R-CNN到DETR,六大经典模型深度对比与实战指南》
人工智能·yolo·目标检测·r语言·cnn
Shang1809893572620 小时前
T41LQ 一款高性能、低功耗的系统级芯片(SoC) 适用于各种AIoT应用智能安防、智能家居方案优选T41L
人工智能·驱动开发·嵌入式硬件·fpga开发·信息与通信·信号处理·t41lq
Bony-20 小时前
用于糖尿病视网膜病变图像生成的GAN
人工智能·神经网络·生成对抗网络
罗西的思考21 小时前
【Agent】 ACE(Agentic Context Engineering)源码阅读笔记---(3)关键创新
人工智能·算法