ARTICLE DETAIL

资讯详情

深耕网站建设与运营推广的一线实战洞察。

零基础跑通 openpi 机械臂智能控制:4 条命令完成仿真验证

零基础跑通 openpi 机械臂智能控制:4 条命令完成仿真验证 零基础跑通 openpi 机械臂智能控制4 条命令完成仿真验证【免费下载链接】openpi项目地址: https://gitcode.com/GitHub_Trending/op/openpiopenpi 是 Physical Intelligence 发布的开源模型与工具集一套基于 π₀ 等视觉-语言-动作VLA模型的机械臂控制系统给机器人配上相机和一句自然语言指令模型直接输出关节动作序列。本文面向没有硬件的新手用仿真环境走通完整推理链路。数据链路与平台支持情况先看一个具体场景MuJoCo 里的双臂要从传送带把方块放到目标位置任务指令就是一句 transfer the cube。客户端把相机图像、关节状态和指令打包发给策略服务器服务器回传接下来 10 步的关节动作客户端按 50Hz 执行后再取新的观测——这就是 openpi 的完整闭环客户端与服务器的分离是 openpi 的核心设计模型可以跑在远离机器人的 GPU 服务器上机器人端只需要一个依赖很少的 packages/openpi-client/ 包接入方式见 docs/remote_inference.md。目前支持四个机器人平台平台硬件要求现成 checkpoint适用场景ALOHA 仿真MuJoCo无pi0_aloha_sim系统入门、零硬件验证ALOHA 实机双臂 4 个相机pi0_aloha_towel 等实机微调、折叠类任务DROIDFranka机器人 远端 GPU 服务器pi05_droid桌面抓取、移动操作LIBERO无基准环境pi05_libero训练复现、基准评测新手建议从 ALOHA 仿真开始官方示例最完整见 examples/aloha_sim/。最小路径启动 ALOHA 仿真环境一条命令预检 Docker 与 GPU下面这条命令一次性确认 Docker、Compose 可用以及 GPU 驱动正常最后一段没有输出就说明 NVIDIA 驱动有问题docker version docker compose version nvidia-smi⚠️ openpi 要求 rootless 模式 DockerGPU 需要装 NVIDIA Container Toolkitsnap 版 Docker 和 Docker Desktop 都与 NVIDIA runtime 不兼容。Ubuntu 22.04 可以按 docs/docker.md 用仓库自带脚本完成安装。用 compose 拉起两个容器先克隆仓库记得初始化子模块依赖了 aloha 等第三方代码git clone https://gitcode.com/GitHub_Trending/op/openpi cd openpi git submodule update --init --recursive设置服务器环境变量一条命令构建并启动 runtime仿真 客户端和 openpi_server推理服务两个容器export SERVER_ARGS--env ALOHA_SIM docker compose -f examples/aloha_sim/compose.yml up --build首次构建会拉取基础镜像和 Python 依赖体积约 5GB之后的运行会命中缓存。启动成功的三个信号openpi_server 日志出现Creating server (host: ..., ip: ...)并监听 0.0.0.0:8000首次推理触发 checkpoint 自动下载~/.cache/openpi可用OPENPI_DATA_HOME覆盖下出现 pi0_aloha_sim 权重文件runtime 容器弹出 MuJoCo 窗口可见双臂任务循环以 50Hz 推进回放视频被写入data/aloha_sim/videos/容器方案与本地运行对比容器方案本地运行适合谁新手、宿主机已装 ROS 等依赖需要改模型代码、调试推理的人启动方式一条 compose 命令拉起服务端与仿真两个终端MUJOCO_GLegl python examples/aloha_sim/main.py与uv run scripts/serve_policy.py --env ALOHA_SIM前置条件rootless Docker NVIDIA toolkitGIT_LFS_SKIP_SMUDGE1 uv sync装好依赖本地模式如遇 EGL 报错通常是缺图形库执行sudo apt-get install -y libegl1-mesa-dev libgles2-mesa-dev即可。推理时延诊断与高频排错时延诊断命令另开一个终端跑无机器人客户端向服务器连发 100 步随机观测最后输出一张分位数时延表Mean/Std/P50/P95指标拆分为 client_infer_ms 和 server_* 两类uv run examples/simple_client/main.py --env ALOHA_SIM --num_steps 100P95 明显高于 P50 时先查服务器 GPU 占用和网络client_infer_ms 整体偏高则检查客户端到服务器之间的网段远程部署时建议走有线网络。三条高频问题症状容器内报 CUDA 或 GPU 错误容器里nvidia-smi异常根因未安装 NVIDIA Container Toolkit或正在用 snap 版 Docker / Docker Desktop修复运行 scripts/docker/install_nvidia_container_toolkit.sh并把 Docker 换成 rootless 版本症状uv sync反复报依赖冲突根因uv 版本过旧或残留旧 .venv修复uv self update后删除 .venv重新执行GIT_LFS_SKIP_SMUDGE1 uv sync症状客户端连不上 8000 端口根因推理服务尚未就绪checkpoint 还在下载或端口/防火墙不通修复先看服务端日志是否已打印监听信息再从客户端侧ping server_ip验证可达性容器模式下两服务共享 host 网络多数情况只是启动顺序问题继续深入自训数据与远程推理跑通推理后下一步通常是让模型适配你的任务与硬件用自己的数据微调先转成 LeRobot 格式参考 examples/libero/convert_libero_data_to_lerobot.py再依次执行uv run scripts/compute_norm_stats.py --config-name 配置名和uv run scripts/train.py 配置名 --exp-name...服务自己训练的 checkpointuv run scripts/serve_policy.py policy:checkpoint --policy.config... --policy.dir...使用 PyTorch 实现src/openpi/models_pytorch/ 与scripts/train_pytorch.pyJAX 权重可用examples/convert_jax_model_to_pytorch.py转换把远程推理嵌进自有机器人代码装openpi-client用WebsocketClientPolicy.infer()构造观测字典训练配置与归一化细节src/openpi/training/config.py和 docs/norm_stats.md硬件参考单卡推理显存 8GB 以上LoRA 微调 22.5GB 以上全参微调 70GB 以上对应 RTX 4090 / A100。 环境验证通过后按下面三条命令依次执行即可完成首次运行git clone https://gitcode.com/GitHub_Trending/op/openpi cd openpi export SERVER_ARGS--env ALOHA_SIM docker compose -f examples/aloha_sim/compose.yml up --build【免费下载链接】openpi项目地址: https://gitcode.com/GitHub_Trending/op/openpi创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表