在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

相关推荐
Together_CZ4 小时前
YOLO-IOD: Towards Real Time Incremental Object Detection——迈向实时增量目标检测
yolo·目标检测·yolo-iod·迈向实时增量目标检测·增量检测·incremental·real time
YOLO数据集集合4 小时前
固定翼无人机目标检测数据集 | 固定翼无人机 低空安防 目标检测 YOLO格式9062期
yolo·目标检测·无人机·固定翼无人机·固定翼·无人机识别·反无人机
RobinDevNotes4 小时前
模型训练工程师必须搞懂的DDP和FSDP
人工智能·pytorch
洛阳纸贵4 小时前
AI-PyTorch(一)基础代码实操和自动求导
人工智能·pytorch·python
磁场转动100万匹6 小时前
PyTorch 食物图像分类实战:从数据准备到 CNN 训练全流程解析
pytorch·分类·cnn
深度学习lover6 小时前
<数据集>番茄叶片病害识别<目标检测>
人工智能·yolo·目标检测·计算机视觉·数据集·番茄叶片病害识别
ai小陈6 小时前
PyTorch梯度累积与裁剪实战:小显存也能稳定训练大批次
人工智能·pytorch·python·深度学习·ai·gpu算力
C^h15 小时前
pytorch 适合初学者 0基础学习
人工智能·pytorch·python
论文复现现场1 天前
本地 PyTorch 训练 OOM,第一次租 RTX 4090 云 GPU 怎么迁移项目?从环境检查到 100 Step 跑通
人工智能·pytorch·python·深度学习·cuda
2601_962099081 天前
在Python中使用LSTM和PyTorch进行时间序列预测
pytorch·机器学习·lstm·时间序列预测·数据预处理