资讯动态

零基础上手 openpi:没有机械臂如何跑通机器人策略推理

发布时间:2026/9/12 7:12:13 来源:尧图企业网站定制
零基础上手 openpi没有机械臂如何跑通机器人策略推理【免费下载链接】openpi项目地址: https://gitcode.com/GitHub_Trending/op/openpi不用买机械臂也能验证 openpi 机器人策略推理链路。openpi 是 Physical Intelligence 开源的一套视觉-语言-动作VLA模型模型看相机画面、听懂语言指令输出一串机械臂动作。全文 4 个步骤命令均按仓库核实过可直接复制。读完你会得到Ubuntu 22.04 上用 3 条命令装好 openpi 环境的完整记录一套无机械臂的冒烟测试方法客户端会打印每次推理的耗时统计出数字即链路通了3 种部署场景的选型表以及 4 个高频报错的修复清单。openpi 到底是干什么的跑之前先弄懂 3 个事实这一节回答“这个项目是什么、我的机器能不能跑”看完你就有判断依据。事实一仓库里是 3 个模型加一套可用权重。π₀ 是基于流匹配的 VLA 模型π₀-FAST 是基于动作 tokenizer 的自回归 VLAπ₀.₅ 是 π₀ 的增强版。基座权重在 1 万小时机器人数据上预训练各平台ALOHA、DROID、LIBERO还有微调好的 checkpoint下载即可用。事实二跑推理一块 8GB 以上显存的 NVIDIA 显卡就够。官方给出的估算推理需要 8GB 以上显存RTX 4090 没问题微调要求更高LoRA 要 22.5GB 以上全参微调要 70GB 以上得用 A100 80GB 或 H100。事实三目前只支持 Ubuntu 22.04。主实现基于 JAX2025 年 9 月起新增 PyTorch 版本覆盖 π₀ 和 π₀.₅。如果你的机器暂时不满足可以先只看推理部分。在 Ubuntu 22.04 上安装 openpiuv 三步走这一节解决依赖问题做完后项目里会有一套可直接用的虚拟环境。openpi 用 uv一个比 pip 快得多的 Python 包管理器管理依赖先确认uv --version能跑通。第一条命令获取代码。third_party/目录下的平台代码是子模块--recurse-submodules不能省git clone --recurse-submodules https://gitcode.com/GitHub_Trending/op/openpi cd openpi成功后你会看到完整的 openpi 目录。接下来两条命令把依赖装进虚拟环境。前缀GIT_LFS_SKIP_SMUDGE1是仓库拉取 LeRobot 依赖时的要求不要删GIT_LFS_SKIP_SMUDGE1 uv sync GIT_LFS_SKIP_SMUDGE1 uv pip install -e .没有报错即安装成功。系统环境已经一团糟的话可以直接跳到官方 Docker 镜像要求和坑都写在 docs/docker.md 里。没有机械臂的第一次运行simple_client 冒烟测试 做完这节你得到的是一份推理链路“体检报告”只要打印出耗时统计表安装、权重、网络全通。冒烟测试的思路是examples/simple_client/脚本生成随机图像和随机关节状态发给服务器服务器用真模型跑推理。默认 ALOHA_SIM 环境对应官方pi0_aloha_sim权重首次运行会自动下载并缓存到~/.cache/openpi第一次会慢一些。终端 1启动策略服务器就绪后监听 8000 端口uv run scripts/serve_policy.py --env ALOHA_SIM终端 2运行客户端默认跑 20 步可用--num-steps调整步数uv run examples/simple_client/main.py --env ALOHA_SIM预期看到一张 “Timing Statistics” 表行是client_infer_ms客户端每次调用耗时和服务器端各计时项列是 Mean、P50、P99 等统计量。数字持续输出说明链路已通。工作原理一张图加一句话如何变成一串动作看懂这条数据流你以后才知道真机械臂该从哪里接进来。openpi 采用客户端-服务器分离客户端可以是机器人程序、仿真器或冒烟脚本只有服务器需要加载模型。三个要点观测图像、状态、prompt由客户端经 websocket 发出模型返回的不是单个动作而是一整段 action chunk机器人端开环执行其中前 N 步耗尽再查一次——这就是模型不需要每秒实时应答的原因。把客户端嵌入自己的机械臂程序怎么写见 docs/remote_inference.md。按场景选部署方式冒烟测试、可视化仿真、真机三条路表格是重点看完你就知道自己的目标该用哪组命令场景运行内容硬件要求启动方式适用时机无机械臂冒烟测试simple_client 策略服务器一块 NVIDIA GPU显存 8GB上一节的两条命令第一次跑验证环境与权重可视化仿真无真机aloha_simMuJoCoGPU 显示器docker compose -f examples/aloha_sim/compose.yml up --build想亲眼看到机械臂动起来真机 远程推理DROID 模式服务器 4090 级 GPU机器人 PC 可低配uv run scripts/serve_policy.py --envDROID机器人电脑没有好显卡想亲眼看到机械臂动aloha_sim 的 Docker 方式一次拉起“仿真 服务器”export SERVER_ARGS--env ALOHA_SIM docker compose -f examples/aloha_sim/compose.yml up --build首次运行要构建镜像耐心等待成功后会弹出 MuJoCo 窗口显示机械臂服务器容器同时带着 ALOHA_SIM 参数启动。不用 Docker 也可以双终端终端 1 按仓库说明建好 venv 后运行MUJOCO_GLegl python examples/aloha_sim/main.py终端 2 跑与冒烟测试相同的serve_policy.py --env ALOHA_SIM。DROID 远程模式对真机最有价值模型放在好 GPU 的服务器上机器人 PC 只装packages/openpi-client/这个轻量客户端包然后指向服务器 IP# GPU 服务器上 uv run scripts/serve_policy.py --envDROID # 机器人 PC 上 python3 examples/droid/main.py --remote_host服务器IP --external_cameraleft官方提示每个动作块 0.5~1 秒延迟属正常机器人端建议走有线网络。跑不起来时4 个高频报错场景及修复⚠️ 以下都来自仓库的 troubleshooting 区按出现概率排序。1.uv sync报依赖冲突原因旧虚拟环境与锁定文件冲突。修复删除.venv目录后重跑uv sync仍报错就先升级 uv 自身。2. 服务器已启动客户端连不上原因服务器还没就绪或两端端口不一致。修复默认端口是 8000确认服务器进程存活远程时检查防火墙与网络连通。3. aloha_sim 窗口空白或报 EGL 错误原因图形库缺失。修复apt 安装libegl1-mesa-dev和libgles2-mesa-dev或把渲染后端MUJOCO_GL换成 glfw 再试。4. 训练时显存不足原因JAX 默认只用 75% 显存。修复启动前设置XLA_PYTHON_CLIENT_MEM_FRACTION0.9全参微调请先对照 70GB 显存要求选卡。用自己的任务微调 openpi三步数据流 这一节是路线图而非完整教程官方以 LIBERO 数据集为示范。①把数据转成 LeRobot 格式官方提供了可直接改的示例脚本见 examples/libero/convert_libero_data_to_lerobot.py。 ②算统计量并启动训练先跑uv run scripts/compute_norm_stats.py --config-name pi05_libero计算归一化统计跳过会报 missing norm stats再跑uv run scripts/train.py pi05_libero --exp-namemy_experimentcheckpoint 存入checkpoints/目录。 ③把新权重挂上服务器uv run scripts/serve_policy.py policy:checkpoint --policy.configpi05_libero --policy.dircheckpoints/pi05_libero/my_experiment/20000客户端按冒烟测试的方式接入即可。偏好 PyTorch 的话入口是scripts/train_pytorch.pyJAX 权重可用examples/convert_jax_model_to_pytorch.py转换。从 clone 到冒烟测试出耗时表你已经跑通了 openpi 的推理链路模型是大脑服务器是躯干客户端可以是一切。现在打开终端先跑一条uv run scripts/serve_policy.py --env ALOHA_SIM。下一步从接上你的机械臂开始。【免费下载链接】openpi项目地址: https://gitcode.com/GitHub_Trending/op/openpi创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

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

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

免费获取报价