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模型,通过使能融合算子、图模式、计算逻辑优化等手段,实现了较低的推理时延。

Pi0昇腾Atlas A2系列适配及优化代码仓链接

Logo

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐