资讯动态

openpi 机器人智能体快速上手指南:3步跑通 VLA 模型推理,4步微调部署

发布时间:2026/9/12 14:16:35 来源:尧图企业网站定制
openpi 机器人智能体快速上手指南3步跑通 VLA 模型推理4步微调部署【免费下载链接】openpi项目地址: https://gitcode.com/GitHub_Trending/op/openpiopenpi 是 Physical Intelligence 开源的机器人智能体项目内置 π₀、π₀-FAST、π₀.₅ 三种 VLA 模型。本篇带你选模型、装环境、三步跑通推理再用自己数据微调并部署。它解决什么问题VLA 模型是什么把它理解成机器人的眼睛、耳朵加双手看相机图像、听语言指令最后输出一串关节动作。openpi 把这类模型做成了开源、可直接微调的工程包覆盖三类典型场景真实桌面操作DROID Franka 专家模型能在新桌面场景里零样本完成拿叉子这类任务ALOHA 双臂专家模型叠毛巾、开食品盒拿到即跑。仿真基准评测想量化模型能力可以跑 LIBERO 基准π₀.₅-LIBERO 检查点在该基准上取得 SOTA 成绩。远程控机模型留在 GPU 服务器通过 WebSocket 把动作流推给机器人端本体不用装重型依赖。三个场景背后是同一套资产3 种模型结构每类都提供经 10,000 小时机器人数据预训练的基础检查点以及针对 DROID、ALOHA、LIBERO 微调好的专家检查点。先选对模型先看架构再看检查点。openpi 提供 3 种模型结构π₀基于流匹配一种扩散式方法生成动作轨迹推理表现稳定是求稳的选择。π₀-FAST基于 FAST 动作标记器的自回归模型像语言模型一样逐个吐动作 token速度更快。π₀.₅π₀ 的升级版用知识绝缘技术训练——类比给主干知识加一层隔离微调新任务时不冲掉预训练能力开放世界泛化更强。基础检查点全部经 10,000 小时机器人数据预训练适合作为微调起点专家检查点已针对具体平台和任务调好可直接推理。选型对照表如下模型适合场景直接推理 / 微调起点数据规模π₀base通用桌面操作求稳微调起点10,000 小时预训练π₀-FASTbase对推理延迟敏感的任务微调起点10,000 小时预训练π₀.₅base开放环境、复杂指令跟随微调起点10,000 小时 知识绝缘训练π₀-FAST-DROID / π₀-DROIDDROID 真机桌面操作直接推理后者可再微调DROID 数据集微调π₀-ALOHA 系列叠毛巾/开饭盒ALOHA 双臂固定任务直接推理ALOHA 内部数据微调π₀.₅-LIBERO / π₀.₅-DROIDLIBERO 仿真基准 / DROID 真机直接推理LIBERO、DROID 数据集微调怎么挑只想先看效果直接选 DROID 或 LIBERO 的专家模型有自己的任务数据就从对应 base 检查点开始微调环境越开放、指令越复杂越建议选 π₀.₅ 系。✅从安装到跑通第一步环境要求Ubuntu 22.04、NVIDIA GPU推理 8GB 起步LoRA 微调 22.5GB 起、已装好 uv 的机器。系统环境不好搞的话可以看 docs/docker.md 的 Docker 方案。三步走拉代码记得带上子模块git clone --recurse-submodules https://gitcode.com/GitHub_Trending/op/openpi装依赖GIT_LFS_SKIP_SMUDGE1是拉取 LeRobot 依赖的必要开关cd openpi GIT_LFS_SKIP_SMUDGE1 uv sync GIT_LFS_SKIP_SMUDGE1 uv pip install -e .跑第一次推理没有机器人也能测这个客户端会构造随机观测并打印推理帧率uv run examples/simple_client/main.py --env DROID 看到动作数字和推理速率输出说明整条链路已通。想看代码版示例参考 examples/inference.ipynb。进阶玩法训练你自己的机器人微调链路是数据转换 → 统计计算 → 启动训练 → 部署策略服务下面以 π₀.₅ LIBERO 为例。① 转换数据为 LeRobot 格式openpi 的训练只认 LeRobot 数据集格式先把自己的数据转过去纯 LIBERO 微调可跳过官方已预转换uv run examples/libero/convert_libero_data_to_lerobot.py --data_dir /path/to/your/data转换脚本很短照着 examples/libero/convert_libero_data_to_lerobot.py 改成你自己的数据源即可。② 计算归一化统计训练前模型需要知道状态/动作的分布来做归一化尺度不对损失就会发散细节见 docs/norm_stats.md。用你的训练配置名跑一遍uv run scripts/compute_norm_stats.py --config-name pi05_libero③ 启动微调训练在 src/openpi/training/config.py 确认数据处理与超参LIBERO 示例配置注释很详细数据到模型的映射在 src/openpi/policies/libero_policy.py。前面的0.9让 JAX 用到 90% 显存XLA_PYTHON_CLIENT_MEM_FRACTION0.9 uv run scripts/train.py pi05_libero --exp-namemy_experiment --overwrite④ 部署策略服务训练完把检查点挂到策略服务器上完成 π₀ 推理部署服务默认监听 8000 端口等待机器人端发观测uv run scripts/serve_policy.py policy:checkpoint --policy.configpi05_libero --policy.dircheckpoints/pi05_libero/my_experiment/20000两个扩展方向 WebSocket 远程推理模型部署到远程 GPU 服务器机器人端只装轻量的 openpi-client观测发过去、动作流回来。机器人端不用装重型依赖换 GPU 也不用动机器人代码完整用法见 docs/remote_inference.md。 PyTorch 生态π₀ 与 π₀.₅ 现在提供 PyTorch 实现已在 LIBERO 上验证推理与微调π₀-FAST、LoRA 暂不支持。JAX 检查点可以转成 PyTorch 格式之后走同一套 API多 GPU 训练用 torchrun 起 DDPuv run examples/convert_jax_model_to_pytorch.py --checkpoint_dir /path/to/jax/checkpoint --config_name pi05_libero --output_path /path/to/converted uv run torchrun --standalone --nnodes1 --nproc_per_node2 scripts/train_pytorch.py pi0_aloha_sim --exp_name pytorch_ddp_test避坑指南现象解法uv sync报依赖冲突删掉.venv目录后重跑uv sync仍失败就uv self update升级 uv训练显存不足 (OOM)命令前加XLA_PYTHON_CLIENT_MEM_FRACTION0.9多卡时用--fsdp-devices n开启分片并行训练启动报缺 norm stats先用scripts/compute_norm_stats.py带上配置名算好归一化统计再启动训练openpi 的路线还在加速更多机器人平台适配、PyTorch 特性补齐都在推进中。挑一个 DROID 专家模型今天就把第一次推理跑通下周轮到你的数据集上场。【免费下载链接】openpi项目地址: https://gitcode.com/GitHub_Trending/op/openpi创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

读完文章,也想定制专属网站?

尧图设计师 24 小时内与您沟通定制方案

免费获取报价