模型部署
文章平均质量分 94
我宿孤栈
略略略~
展开
-
推理模型部署-训练好的权重文件如何生成.exe文件在另一台电脑运行
部署大型语言模型是一个详尽且复杂的过程,涵盖了从数据处理到模型优化、集成和维护的多个环节。通常模型训练好之后,在源码里边也有detect.py代码进行推理,但是如何将训练好的模型给下游使用或者部署到生产环境中而不使用源码呢?模型部署后,我们需要对模型的性能进行评价,以确保模型能够在生产环境中高效地运行。模型部署是机器学习应用中重要的一环,它可以帮助企业快速将机器学习应用落地,并带来实际的商业价值。总之,Python中有很多方法可以实现模型部署,需要根据具体的需求选择适合的方法。模型部署相关的上下游模块。原创 2024-04-24 15:05:16 · 1184 阅读 · 0 评论 -
VS编译部署libtorch-yolov5推理运行自己训练的权重文件/模型(CPU和GPU版本)
前期环境配置(vs+libtorch+opencv)可以参考博主另一篇博文,这里主要基于环境配置好之后如何运行yolov5的推理程序,并生成对应的.exe文件。libtroch版本尽量和pytorch的版本一致,。博主查阅了很多博文,有的博文是通过cmake编译运行,博主vs新手,因为VS配置的libtorch和opencv是跟着项目的(在vs界面),博主暂时还没能把终端cmake编译和vs联系起来(配置好的环境容易崩掉),后边有时间再出一版结合cmake编译运行的方案。原创 2024-04-17 16:59:35 · 1772 阅读 · 1 评论 -
win下C++通过Clion部署yolov5——libtorch+yolov5
其中【images】和【weights】即为二中下载源码中的【images】和【weights】,其他的头文件和源文件是二中源码。蓝色框则是添加需要运行的头文件和源文件,这里就是和博文。中的cmakelist.txt的不同之处,所以关于红色框和绿色框的路径地址怎么给,也可以参考该博文。在运行之前,需要将【images】复制到编译目录下,这是方便自定义给测试图片时路径书写方便。这里博主没有新建【include】和【src】目录,直接将5个文件复制到了新建项目下。对于终端会闪退的问题,可以添加如下代码解决。原创 2023-05-17 10:15:06 · 1114 阅读 · 0 评论 -
win下pytorch安装—cuda11.6 + cudnn8.4 + pytorch1.12 + tensorRT(pycuda)
下载解压后将bin以及include文件夹复制粘贴到步骤一种cuda安装目录下,同cudnn,直接复制粘贴会自动合并;这里选择默认目录即可,是个安装过程的临时目录,结束后此文件夹会被自动删除。需要注意的是这里并非指定安装目录,若需要指定安装目录,此处需与指定安装目录不同。注意上图的红色框框,是因为博主环境变量配置的问题。正确配置后即可查看安装的tensorRT版本了。这里有个前置工作需要搞清楚的,就是自己的显卡支持的cuda版本。目录下两个exe文件后是否出现跑“pass”,出现即为安装成功。原创 2023-03-22 19:17:18 · 5960 阅读 · 0 评论 -
学习笔记之OpenMMLab-—模型部署基础知识
torch.onnx.export()得到onnx模型;netron.app查看onnx结构(onnx整体结构、算子属性信息和输入输出信息);了解算子集(http://onnx.ai/onnx/operators/)原创 2023-03-21 16:57:11 · 1143 阅读 · 0 评论