4步跑通openpi:用Docker搭建VLA机器人模型运行环境
2026/9/20 21:29:01 网站建设 项目流程

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(官方只测试了该系统)
GPUNVIDIA 显卡,推理需 >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 docker

3️⃣ 任务三:给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/,它是最小客户端。

验证:客户端日志持续打印观测输入和推理帧率,没有断连报错。

🔍 排障自查

遇到卡住时,按"现象 → 先查什么 → 怎么处理"对号入座:

现象先查什么怎么处理
dockerpermission denied重启后是否重新登录过(用户组要重登录才生效)groups \| grep docker确认在组内;不在则sudo usermod -aG docker $USER,注销重登
容器报找不到 GPU /libnvidia-ml.so是否装着 snap 版 Docker 或 Docker Desktop(二者与 NVIDIA 工具包冲突)sudo snap remove dockersudo 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),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询