Skip to main content

Jetson 上的 Microduck 强化学习

Microduck reinforcement learning on Jetson

本演示在一台由 NVIDIA Jetson Orin NX 16GB 驱动的 Seeed reComputer 上,为 Microduck 构建了一套完整的机器人学习工作流。内容涵盖 GPU 环境部署、基于 MuJoCo 的 PPO 训练、本地检查点可视化、使用官方 ONNX 策略的键盘控制推理,以及新自定义动作的开发。

经验证的参考平台使用 JetPack 7.2Ubuntu 24.04CUDA 13.2Python 3.12MuJoCo 3.10。本教程基于 jjjadand/microduck_rl 仓库,该仓库包含本指南中使用的源代码、部署脚本、官方 ONNX 策略以及在 Jetson 上训练得到的检查点。

硬件选项

本教程在 Jetson Orin NX 16GB 上复现。以下 Seeed Studio 系统提供了适合跟随本工作流的 Jetson 平台。reComputer Classic J5011 提供了更高的 GPU 和内存容量,而实际训练吞吐量取决于所选功耗模式和并行环境数量。

reComputer Super J4012 with Jetson Orin NX 16GB
reComputer Super J4012
NVIDIA Jetson Orin NX 16GB
reComputer Classic J5011 with Jetson AGX Orin 32GB
reComputer Classic J5011
NVIDIA Jetson AGX Orin 32GB

选择章节

点击卡片打开对应章节。Physical AI 侧边栏中只列出此落地页;三个章节保持为从这里访问的聚焦页面。

你将复现的内容

  • 在 Jetson Orin NX 上启用 CUDA 的 Microduck 训练环境。
  • 使用并行 MuJoCo 环境的 PPO 训练。
  • 原生和基于浏览器的仿真可视化。
  • 针对行走、站立、坐/站、地面拾取、翻滚、踢腿和滚轮动作的官方 ONNX 推理。
  • 键盘指令输入和实时行为切换。
  • 一套可复用的工作流,用于创建如鞠躬等基于阶段的自定义动作。

演示架构

Jetson Orin NX
├── JetPack 7.2 / CUDA 13.2
├── uv project environment
├── mjlab + MuJoCo Warp + rsl_rl
├── Microduck task configurations
├── PPO checkpoint (.pt)
└── Exported policy (.onnx)
├── MuJoCo inference rehearsal
└── Microduck runtime deployment
tip

为了最快速地完成验证,请先部署环境,运行 64 个环境、5 次迭代的冒烟测试,然后启动官方的 ONNX 键盘演示。你可以在之后再完成自定义动作章节。

Loading Comments...