4步跑通openpi:用Docker搭建VLA机器人模型运行环境
【免费下载链接】openpi项目地址: https://gitcode.com/GitHub_Trending/op/openpi
openpi 是 Physical Intelligence 团队开源的 VLA(视觉-语言-动作)机器人模型仓库,内置 π₀、π₀-FAST、π₀.₅ 三类模型,开箱即可推理,也能微调到自有数据集。按本文做完,你的电脑上会有两个正常运行的容器:一个策略推理服务器(policy server),一个 aloha_sim 仿真客户端。成功标准很具体——客户端日志持续打印推理帧率,说明模型已在你的 GPU 上跑通。
下面用 Docker 部署而不是裸装依赖,原因一句话:它帮你绕开 ROS 等大型依赖,环境一致、不污染系统。
📋 准备清单
开始前核对一遍,全部满足再动手:
| 项目 | 要求 |
|---|---|
| 系统 | Ubuntu 22.04 LTS(官方只测试了该系统) |
| GPU | NVIDIA 显卡,推理需 >8GB 显存(官方示例参考 RTX 4090) |
| 磁盘 | 20GB 以上可用空间(镜像 + 模型缓存) |
| Docker | 需要能跑 rootless 模式;snap 安装的 Docker 和 Docker Desktop 与 NVIDIA 容器工具包不兼容,二选一都没有才继续 |
1️⃣ 任务一:克隆仓库
目标:拿到完整源码,包括 aloha、libero 等子模块(缺少子模块会导致示例构建失败)。
操作:一条命令克隆并递归拉取子模块:
git clone --recurse-submodules https://gitcode.com/GitHub_Trending/op/openpi验证:进入目录后能看到scripts/docker/、examples/、third_party/aloha等目录,third_party下不是空文件夹。
2️⃣ 任务二:一键装好Docker
目标:装好 Docker Engine 并配置好当前用户权限,之后跑docker命令不用sudo。
操作:仓库自带安装脚本,一条命令完成加源、装包、加用户组、设开机自启:
bash scripts/docker/install_docker_ubuntu22.sh脚本跑完会提示重启系统——必须重启,否则用户组权限不生效。
验证:重启后执行下面命令,输出里出现 docker 组即成功:
groups | grep docker3️⃣ 任务三:给Docker接上NVIDIA显卡
目标:让容器能访问 GPU,这是模型推理能跑起来的硬前提。
操作:同样用仓库自带的脚本,它负责加 NVIDIA 软件源、装nvidia-container-toolkit、配置 Docker 运行时:
bash scripts/docker/install_nvidia_container_toolkit.sh验证:脚本末尾会自动重启 Docker 服务。装好后可先跑一个最小检查:
docker run --rm --gpus all nvidia/cuda:12.2.2-base nvidia-smi输出显卡型号和显存信息即 GPU 通道打通。
4️⃣ 任务四:构建基础镜像并启动策略服务器
目标:构建openpi_server镜像并启动它,模型服务在后台监听,等待客户端连入。
操作:在仓库根目录执行:
docker compose -f scripts/docker/compose.yml up --build首次构建要拉 CUDA 基础镜像并同步 Python 依赖,需要 10~20 分钟;之后镜像有缓存,启动会快很多。
验证:终端停在服务器日志处不再退出,且:
docker ps能看到openpi_server容器处于Up状态。
如果你的机器没有NVIDIA 显卡,需要先把 scripts/docker/compose.yml 里的
deploy(GPU 申请)代码块注释掉再启动,否则容器会因申请不到 GPU 而起不来。
5️⃣ 任务五:跑一个能看见结果的示例
目标:用 aloha_sim 仿真客户端把闭环走通——它向策略服务器发观测,模型返回动作,仿真画面动起来的瞬间就是跑通。
操作:--env告诉服务器加载哪个环境的配置,然后启动 compose(服务器 + 仿真器一起拉起):
export SERVER_ARGS="--env ALOHA_SIM" docker compose -f examples/aloha_sim/compose.yml up --build不想看仿真画面、只想验证推理链路的,可以换 examples/simple_client/,它是最小客户端。
验证:客户端日志持续打印观测输入和推理帧率,没有断连报错。
🔍 排障自查
遇到卡住时,按"现象 → 先查什么 → 怎么处理"对号入座:
| 现象 | 先查什么 | 怎么处理 |
|---|---|---|
跑docker报permission denied | 重启后是否重新登录过(用户组要重登录才生效) | groups \| grep docker确认在组内;不在则sudo usermod -aG docker $USER,注销重登 |
容器报找不到 GPU /libnvidia-ml.so | 是否装着 snap 版 Docker 或 Docker Desktop(二者与 NVIDIA 工具包冲突) | 用sudo snap remove docker或sudo apt remove docker-desktop移除,改回 apt 版 Docker |
| compose 启动即退、日志有 GPU 相关报错 | 有没有 NVIDIA 卡却保留了 GPU 申请配置 | 注释掉 compose 文件中的deploy块再启动 |
构建慢通常是网络问题,可换 Docker 镜像源后重新--build,镜像层有缓存,不会从头再来。更多细节见 docs/docker.md。
下一步
- 打开 examples/ 目录,aloha_real(真机)、droid、libero 各有独立 compose 文件和 README,换环境跑法与本文一致;
- 想直接用预训练 checkpoint 做推理而不仿真,参考 examples/inference.ipynb;
- 远程推理(模型跑在另一台 GPU 服务器上)的接法,读 docs/remote_inference.md。
【免费下载链接】openpi项目地址: https://gitcode.com/GitHub_Trending/op/openpi
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考