Pi0机器人VLA具身智能大模型在昇腾Atlas A2系列上已支持推理,跑进80毫秒,代码已开源。欢迎体验,多多star / fork / 转发!
Pi0机器人VLA具身智能模型在昇腾Atlas A2系列上已支持推理,跑进80毫秒。代码已经开源至cann-recipes-embodied-intelligence仓库,旨在针对具身智能业务中的典型模型、加速算法,提供基于CANN平台的优化样例,方便开发者简单、快速、高效地基于CANN平台使用具身智能模型。
·
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模型,通过使能融合算子、图模式、计算逻辑优化等手段,实现了较低的推理时延。 |
昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链
更多推荐

所有评论(0)