恒美微站
首页
关于我们
建站服务
主题模板
案例展示
资讯中心
联系我们
openpi 环境搭建:3 步跑通 π₀ 模型的第一次推理
首页
资讯中心
/
openpi 环境搭建:3 步跑通 π₀ 模型的第一次推理
openpi 环境搭建:3 步跑通 π₀ 模型的第一次推理
发布时间:2026/9/12 13:59:58
openpi 环境搭建3 步跑通 π₀ 模型的第一次推理【免费下载链接】openpi项目地址: https://gitcode.com/GitHub_Trending/op/openpi你刚把 openpi——Physical Intelligence 团队开源的视觉-语言-动作VLA机器人模型仓库——clone 到本地打开根目录模型定义、示例和依赖清单挤在一起而你最想确认的是这台机器到底能不能跑起来 π₀ 的推理。下面走装依赖、起服务、发随机观测这条最短路径走完你就能看到推理耗时统计打印在终端里。环境体检先花一分钟核对机器配置对照 README 中的官方要求检查项要求说明操作系统Ubuntu 22.04官方唯一测试过的系统其他发行版不保证支持Python≥ 3.11uv 会自动创建匹配的虚拟环境显存≥ 8 GB单卡推理的最低门槛RTX 4090 级别即可显存微调LoRA ≥ 22.5 GB全量 ≥ 70 GB全量微调需要 A100 (80GB) / H100磁盘≥ 20 GB 空闲权重与依赖较大检查点缓存在~/.cache/openpi不满足时的降级方案系统不是 22.04 可以直接走 Docker 路线官方镜像基于 Ubuntu 22.04 CUDA 12.2显存不够时推理仍可用多卡并行分摊训练则在配置里设置fsdp_devices做模型并行。最短路径安装第 1 步拿到完整源码。clone 时必须带上子模块否则third_party/下的 ALOHA、LIBERO 依赖是空的。git clone --recurse-submodules https://gitcode.com/GitHub_Trending/op/openpi cd openpi第 2 步装好 uv 并同步依赖。按官方文档先装好 uv然后在仓库根目录执行。GIT_LFS_SKIP_SMUDGE1这一项不能省它是为了以 git 依赖方式拉取 LeRobot。GIT_LFS_SKIP_SMUDGE1 uv sync GIT_LFS_SKIP_SMUDGE1 uv pip install -e .跑完后仓库根目录会多出一个.venv/这就是你的运行环境。第 3 步可选Docker 兜底。如果 uv 安装反复冲突或不想让 ROS 类依赖污染系统改用官方容器docker compose -f scripts/docker/compose.yml up --build构建脚本位于 scripts/docker/细节见 docs/docker.md。跑通第一个 Demo不用机器人的最小示例是 examples/simple_client/main.py它自己生成随机观测发给推理服务并统计每步耗时正好覆盖服务、模型、权重下载整条链路。终端 1启动策略服务。首次运行会自动从远端下载pi05_droid检查点并缓存到~/.cache/openpi看到日志出现Creating server (host: ...)即代表 8000 端口就绪。uv run scripts/serve_policy.py --env DROID终端 2发送观测。默认参数就会连接 8000 端口的服务跑 20 步推理。uv run examples/simple_client/main.py --env DROID判定标准很具体客户端先打印一行Server metadata: ...接着Running policy进度条走完 20/20最后输出标题为Timing Statistics的表格含client_infer_ms、server_infer_ms等行和 mean / P25 / P50 / P99 列。看到这张表说明你的 GPU、权重、网络全链路都通了。故障自救手册症状排查命令解法uv sync报依赖冲突uv --version删除.venv后重跑uv sync仍失败则uv self update升到最新版首次运行报 CUDA / GPU 错误nvidia-smi确认驱动正常即可系统级 CUDA 库不需要装uv 会处理反而可能因系统库冲突而需要卸载docker 命令提示 permission deniedgroups | grep docker输出不含 docker 时执行sudo usermod -aG docker $USER注销重登生效容器里识别不到 GPUdocker info | grep -i nvidia安装 nvidia-container-toolkitsnap 版 docker 与 toolkit 不兼容需sudo snap remove docker后换 apt 版仓库地图openpi/ ├── src/openpi/ │ ├── models/ # π₀ / π₀-FAST / π₀.₅ 的 JAX 模型定义 │ ├── models_pytorch/ # PyTorch 实现及 transformers 补丁文件 │ ├── policies/ # ALOHA / DROID / LIBERO 平台的输入输出映射 │ ├── serving/ # WebSocket 策略服务端 │ ├── training/ # 训练配置、数据加载与分布式分片 │ └── shared/ # 检查点下载、归一化统计等共享工具 ├── examples/ # 各平台的推理、微调与客户端示例 ├── packages/openpi-client/ # 连接策略服务的客户端库 ├── scripts/ # 训练、起服务、算归一化统计等入口 └── docs/ # Docker、归一化统计、远程推理文档两个值得先记住的入口推理服务是 scripts/serve_policy.py微调训练是 scripts/train.py。环境通了之后下一步建议看 examples/libero/README.md 学微调全流程。卡住了先看 README.md 的 Troubleshooting 表再看 docs/提 issue 前参考 CONTRIBUTING.md。【免费下载链接】openpi项目地址: https://gitcode.com/GitHub_Trending/op/openpi创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考