Pi0机器人VLA具身智能大模型在昇腾Atlas A2系列上已支持推理，跑进80毫秒，代码已开源。欢迎体验，多多star / fork / 转发！
·
cann-recipes-embodied-intelligence仓库旨在针对具身智能业务中的典型模型、加速算法，提供基于CANN平台的优化样例，方便开发者简单、快速、高效地基于CANN平台使用具身智能模型，丰富CANN开源开放生态。
基于LeRobot库，在昇腾Atlas A2环境适配Pi0模型的pytorch版本，通过使能昇腾融合算子、图模式、计算逻辑优化等手段，实现了较低的推理时延，单卡上单次推理时间跑进80毫秒。目前，Pi0模型在昇腾A2上的迁移优化代码已经开源在gitcode代码管理平台（https://gitcode.com/cann/cann-recipes-embodied-intelligence）。欢迎大家多多star加星 / fork / 转发，多交流。
✨样例列表
| 实践 | 简介 |
|---|---|
| Pi0 | 基于LeRobot库，在Atlas A2环境适配Pi0模型，通过使能融合算子、图模式、计算逻辑优化等手段，实现了较低的推理时延。 |
更多推荐
2
0- 0
已为社区贡献2条内容
所有评论(0)