AMD:开源 Schola 在 ROCm 平台上连接 Unreal Engine 与强化学习,用于机器人手臂训练
AMD 推出了 Schola,这是一款用于 Unreal Engine 的开源插件,通过 Python 框架和 gRPC 实现与 Gymnasium 兼容的强化学习训练。示例中,协作机器人手臂 xArm6 在 Unreal Engine 5.7 中配合 MuJoCo 物理引擎、PPO 算法和 PyTorch 在 AMD ROCm GPU 加速栈上进行训练。教程展示了一个到达任务,机械臂末端移动至随机目标位置。
本文由人工智能基于一手来源生成。
AMD 发布了 Schola 教程,这是一款将 Unreal Engine 与强化学习相连接的开源插件,用于在 AMD 硬件上训练机器人。
Unreal 与 RL 框架之间的桥梁
Schola 是一款用于 Unreal Engine 的开源插件,通过 gRPC 连接的 Python 框架实现与 Gymnasium 兼容的**强化学习(RL)**训练。强化学习是一种智能体通过试错最大化奖励来学习的方法。通过将 Unreal 与 Gymnasium 等工具连接,Schola 将逼真的仿真环境与成熟的 RL 库相融合。
在 AMD 栈上训练机器人手臂
示例中,协作机器人手臂 xArm6(六自由度)在 Unreal Engine 5.7 中配合 MuJoCo 物理引擎进行训练。算法采用 Stable Baselines 3 中的 PPO(近端策略优化),计算由 PyTorch 在 AMD ROCm GPU 加速栈上驱动。任务为「到达」任务——机械臂末端移动至随机设定的目标位置。该教程是实践指南,不包含定量基准测试结果,因此作为工具链演示,而非性能比较。
常见问题
- AMD Schola 是什么?
- 一款用于 Unreal Engine 的开源插件,通过 Python 和 gRPC 实现与 Gymnasium 兼容的强化学习训练。
- 示例中训练了什么?
- xArm6 机器人手臂在 Unreal Engine 5.7 中配合 MuJoCo 物理引擎、PPO 算法和 PyTorch 在 AMD ROCm 栈上进行训练。