microduck_rl(MuJoCo+PPO+WandB+UV训练机器人)

MuJoCo:

PPO:

WandB:

UV:

一、训练步骤

Step 1:准备 Ubuntu + NVIDIA GPU

确认能正常看到 GPU:

复制代码
nvidia-smi

Step 2:安装 uv

复制代码
curl -LsSf https://astral.sh/uv/install.sh | sh

Step 3:下载代码

复制代码
git clone https://github.com/pollen-robotics/microduck_rl
cd microduck_rl

Step 4:安装依赖

复制代码
uv sync

Step 5:登录 WandB(登录需要api-key,需要到WandB官网注册账号后创建)

复制代码
wandb login

Step 6:查看任务

复制代码
uv run list-envs

Step 7:跑 smoke test,先确认这个成功。

复制代码
uv run train Mjlab-Velocity-Flat-MicroDuck \
    --env.scene.num-envs 64 \
    --agent.max_iterations 5

Step 8:正式训练

复制代码
uv run train Mjlab-Velocity-Flat-MicroDuck \
    --env.scene.num-envs 4096

官方给出的经验是,4096 environments 下,得到一个可用 gait 大约需要 1--2 小时,具体当然取决于 GPU。

Step 9:查看训练结果

复制代码
uv run play \
    Mjlab-Velocity-Flat-MicroDuck \
    --wandb-run-path <entity/project/run_id>

Step 10:导出

复制代码
uv run scripts/export.py \
    Mjlab-Velocity-Flat-MicroDuck \
    --wandb-run-path <entity/project/run_id>

得到:

复制代码
output.onnx

Step 11:CPU仿真验证

复制代码
uv run scripts/infer_policy.py \
    --walking output.onnx

Step 12:再考虑部署到真实 Microduck

复制代码
output.onnx
      │
      ▼
Microduck policy/runtime
      │
      ▼
robotd
      │
      ▼
真实机器人

二、创建自定义动作

三、创建自定义训练任务

四、创建自定义机器人

五、PPO训练详解(结合cuda+GPU、GPGPU)

相关推荐
Williamtym6 小时前
PickGo Lab实践:用浏览器、Python和MCP串起机器人搬运仿真
开发语言·python·机器人
禁默7 小时前
Embodied VideoAgent 论文解读:机器人怎样持续记住动态场景
人工智能·机器人
szxinmai主板定制专家8 小时前
ZYNQ 数据采集方案踩坑总结:电力强电磁干扰、工业现场多协议适配优化
运维·人工智能·fpga开发·机器人·zynq
空堂与归9 小时前
英伟达开源ASPIRE:机器人自己攒Skill
人工智能·机器人·开源·agent
Gary Studio12 小时前
轮足机器人-IMU部分
机器人
禁默12 小时前
机器人的手,为什么拿个杯子都这么难?一文看懂“灵巧操作“
人工智能·microsoft·机器人
愤怒火龙果14 小时前
张小珺商业访谈笔记整理|对话 Dyna 创始人 Lindon Gao:机器人世界的迷雾、分歧与抉择
人工智能·机器人·人物访谈
具身AGI16 小时前
一块远程显卡服务一队机器人,瓶颈不在模型
人工智能·机器人
波力海苔夹心脆67516 小时前
VisionMaster 常用工具入门:图像采集、定位、测量、识别一篇打通
经验分享·机器人·视觉检测·工业相机·visionmaster·海康
Robert--cao16 小时前
MoE 机器人强化学习原理详解:如何用专家混合表示和 RoboGauge 提升四足机器人 Sim-to-Real 可靠性
机器人