7.2 远程异步推理

7.2 远程异步推理

通过 gRPC 将推理卸载到远程 GPU:

[树莓派] ←→ [GPU 服务器: policy_server]

安装依赖

pip install -e ".[async]"

Step 1: 启动 lekiwi_host(树莓派)

python -m lerobot.robots.lekiwi.lekiwi_host --robot.id R12252801

Step 2: 启动 policy_server(GPU 服务器)

python src/lerobot/scripts/server/policy_server.py --host=0.0.0.0 --port=5858

Step 3: 启动 robot_client(树莓派)

python src/lerobot/scripts/server/robot_client.py \
  --server_address="<GPU_IP>:5858" \
  --policy_type=act \
  --pretrained_name_or_path=outputs/train/act_lekiwi_test/checkpoints/last/pretrained_model \
  --policy_device=cuda \
  --actions_per_chunk=50

LeKiWi 的 private 分支使用旧版 API 路径(src/lerobot/scripts/server/),与新版不同。

启动 host(树莓派)

linux: python -m lerobot.robots.lekiwi.lekiwi_host --robot.id R12252801
macos: python -m lerobot.robots.lekiwi.lekiwi_host --robot.id R12252801
windows: python -m lerobot.robots.lekiwi.lekiwi_host --robot.id R12252801

启动 policy_server(GPU 服务器)

linux: python src/lerobot/scripts/server/policy_server.py --host=0.0.0.0 --port=5858
macos: python src/lerobot/scripts/server/policy_server.py --host=0.0.0.0 --port=5858
windows: python src/lerobot/scripts/server/policy_server.py --host=0.0.0.0 --port=5858