openpi 环境搭建30分钟指南:Docker 3 步从裸机到跑通 VLA 示例
【免费下载链接】openpi项目地址: https://gitcode.com/GitHub_Trending/op/openpi
openpi 是 Physical Intelligence 团队开源的机器人 VLA 模型仓库,提供 π₀、π₀-FAST、π₀.₅ 三个基于 1 万小时机器人数据预训练的基础模型,以及完整的训练、微调与推理流水线。这篇文章只解决一件事:在 Ubuntu 22.04 机器上用 Docker 完成 openpi 环境搭建,跑通一个可交互的示例,全程约 30 分钟。
成果前置:跑通后你会看到什么
环境搭好后,docker ps里应该同时存在两个容器:openpi_server(策略推理服务)和aloha_sim(MuJoCo 仿真),桌面上弹出仿真窗口。
如果手头没有真机器人,可以用仓库自带的 simple client 做冒烟测试——它生成随机观测并打印模型推理速率:
export SERVER_ARGS="--env ALOHA_SIM" docker compose -f examples/simple_client/compose.yml up --build终端持续打印推理帧率,就说明从容器、GPU 到模型推理整条链路已经通了 ✅
环境体检
开工前花 1 分钟对表,任何一项不满足先补齐:
| 检查项 | 要求 |
|---|---|
| 操作系统 | Ubuntu 22.04(官方仅在此版本上测试) |
| GPU 显存 | 推理 ≥ 8 GB;LoRA 微调 ≥ 22.5 GB;全参微调 ≥ 70 GB |
| Docker | rootless 模式安装;snap 版与 Docker Desktop 均与 NVIDIA 运行时冲突 |
| GPU 接入 | 需 NVIDIA container toolkit,否则容器内摸不到显卡 |
操作主线
拉取仓库并初始化子模块
示例依赖third_party/下的子模块(ALOHA、Libero 仿真环境),clone 时必须带上递归参数:
git clone --recurse-submodules https://gitcode.com/GitHub_Trending/op/openpi已经 clone 过的仓库,补一句git submodule update --init --recursive即可。
一键安装 Docker 引擎
仓库内置了 Ubuntu 22.04 专用的安装脚本,GPG 密钥、软件源、docker 用户组、开机自启全部一次做完:
bash scripts/docker/install_docker_ubuntu22.sh脚本末尾会提示重启系统,让 docker 组权限生效。重启后执行docker ps不报权限错误就算通过。
一键安装 GPU 容器工具包
NVIDIA 脚本同样放在仓库里,装完会自动把 runtime 配置进 Docker 并重启服务:
bash scripts/docker/install_nvidia_container_toolkit.sh验证 runtime 是否就位:
docker info | grep -i nvidia有输出即为成功。
构建基础推理服务镜像
用 基础 compose 配置 构建openpi_server镜像,这是后面所有示例共用的策略服务端:
docker compose -f scripts/docker/compose.yml up --build首次构建约 10–20 分钟,之后全部走缓存。没有 GPU 的机器要把 compose 文件里的deploy.resourcesGPU 预留段注释掉再构建。
挂载 aloha_sim 示例并启动
每个示例自带 compose 配置,以 aloha_sim 为例,一条命令同时拉起服务和仿真:
export SERVER_ARGS="--env ALOHA_SIM" docker compose -f examples/aloha_sim/compose.yml up --build构建完成后 MuJoCo 仿真窗口打开、机器人开始执行策略动作,本次环境搭建即可宣告完成 🚀
避坑速查
| 现象 | 快速修复 |
|---|---|
| docker 命令报 permission denied | sudo usermod -aG docker $USER,注销重登 |
| 容器内识别不到 GPU | docker info \| grep -i nvidia查 runtime;确认没装 snap 版 docker 或 Docker Desktop |
| 首次构建卡住很久 | 构建拉取依赖较慢属正常现象,别 Ctrl+C,后续运行走缓存 |
| 推理时显存爆掉 | 8 GB 以下显卡跑不动,换 ≥ 8 GB 卡或走远程推理 |
更完整的背景说明见 Docker 部署文档。
项目地图
openpi/ ├── docs/ # 部署文档:docker、归一化统计、远程推理 ├── examples/ # 各平台示例:aloha_real/aloha_sim/droid/libero/simple_client ├── packages/ # openpi-client 客户端库,含 websocket 连接与图像工具 ├── scripts/ # Docker 安装与构建脚本、训练入口 train.py └── src/openpi/ # 核心源码:models 模型定义、policies 策略封装、training 训练框架延伸路径
环境跑通之后,按这个顺序推进比较顺:
- 无机器人验证:
docker compose -f examples/simple_client/compose.yml up --build,用推理速率确认 GPU 利用率正常。 - 换示例跑:把 compose 路径换成
examples/libero/或examples/droid/,其余命令不变。 - 微调自己的数据:先用 scripts/compute_norm_stats.py 算归一化统计,再执行
uv run scripts/train.py做 LoRA 微调,参考 配置说明。
【免费下载链接】openpi项目地址: https://gitcode.com/GitHub_Trending/op/openpi
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考