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)

相关推荐
米糕闯编程12 小时前
鱼香ros2(二十五)导航
机器人·slam·ros2·launch·navigation2
猫先生Mr.Mao1 天前
RoboRSI详解:机器人自进化为什么需要任务结构和技能边界
机器人·多智能体·论文解读·具身智能·自进化·roborsi
敢敢のwings2 天前
RPV-SemNav 深度解析:让机器人先理解房间,再决定去哪里找目标
机器人
好悬给我拽开线2 天前
RoboJEPA:扩展机器人潜在世界模型
机器人
硅谷秋水3 天前
AgenticRoboRL:将“编程智能体”后训练引入智能体化机器人领域
人工智能·机器学习·语言模型·机器人
敢敢のwings3 天前
SimpleMemVLA 深度解读:把原生视频上下文变成机器人的记忆
数码相机·机器人·音视频
杜子不疼.3 天前
没有真机,我给启元 Q1 造了一台“假机器人“:官方 SDK 示例跑通实录
人工智能·机器人
todoitbo3 天前
启元机器人 Q1 SDK 接口实操:构建 aimdk_msgs 并用 ROS 2 本地 Mock 验证 Service 与 Topic
ai·机器人·二次开发·sdk·启元
lisw053 天前
智能化机器人系统的模块化设计自动化!
机器学习·机器人·自动化
敢敢のwings3 天前
TACDYN-WAM 深度解读:机器人如何预测接触变化,并据此生成动作
人工智能·机器学习·机器人