在jetson thor上高效部署π0.5和yolo模型

在thor上部署π0.5

有2种方法:

  1. 直接使用pytorch + torch.compile
  2. 使用pytorch + tensorrt engine

其中方法2的难点是如何将tensorrt engine与pytorch流程结合起来,

如使用pytorch + action DIT engine或者使用pytorch + llm tensorrt engine + action DIT engine

基本思路是将tensorrt engine调用封装成forward方法并动态替换pytorch实现,这样可以灵活地替换我们想优化性能的部分。再保持高灵活性的同时兼顾推理性能。

具体细节有个生产级的开源项目, 已经完成了量化,编译,部署。十分值得学习和参考:
https://github.com/happyAnger66-an/model_optimizer

相关推荐
YOLO数据集集合2 小时前
UAVDT 无人机车辆检测数据集 - 无人机航拍 | 车辆检测 | 目标检测 | YOLO格式 | 智能交通 | 城市管理 | 多类别数据集 | 计算机视觉
人工智能·yolo·目标检测·机器学习·计算机视觉·目标跟踪·无人机
%4719 小时前
DAY 38
人工智能·pytorch·深度学习
Είναι η κοπέλα1 天前
PyTorch 模型导出与部署实战:ONNX + onnxruntime(可直接落地)
人工智能·pytorch·python
呆萌很1 天前
PyTorch CosineAnnealingLR的T_max和eta_min参数设置
人工智能·pytorch·python
YOLO数据集集合1 天前
火情监测新方案:无人机红外-可见光多模态火点烟雾数据集(含YOLOv11实战)
人工智能·yolo·计算机外设·无人机·无人机数据集·可见光多模态
盼小辉丶1 天前
PyTorch强化学习实战——用预训练语言模型和ChatGPT玩转文本游戏
pytorch·深度学习·语言模型·chatgpt·强化学习
Lee_jerome2 天前
从 PyTorch 权重到 RK3588 板端推理:ResNet18 二分类模型完整部署教程
pytorch·边缘计算·rk3588·模型部署·onnx·resnet18·int8量化
菜冻鱼2 天前
Python-pytorch-高级技巧
开发语言·人工智能·pytorch·python·深度学习·神经网络·聚类
菜冻鱼2 天前
Python-pytorch-模型保存与加载
开发语言·人工智能·pytorch·python·深度学习·机器学习
欧阳天羲2 天前
激光灭蚊机器人项目|雷达‑红外‑YOLO三重融合完整工程
yolo·机器人