强化AI学习 NVIDIA新作将训练速度大幅提升


英伟达NVIDIA为研究人员开发了Isaac Gym高性能机器人模拟平台 。它可以运行一个端到端的GPU加速训练管道,使研究人员能够克服诸多限制,在连续控制任务中实现100倍-1000倍的训练速度 。
【强化AI学习 NVIDIA新作将训练速度大幅提升】Isaac Gym通过将物理缓存中的数据直接传递到 PyTorch 张量进行通信,可以在 GPU 上端到端地实现物理模拟和神经网络策略训练 。  Isaac Gym 提供了一个高性能的学习平台,可以直接在 GPU 上进行各种代理训练 。与使用 CPU 模拟器和 GPU 神经网络的传统 RL 训练相比,Isaac Gym 大大减少了单个 GPU 上复杂机器任务的训练时间,其训练速度提高了 1-2 个数量级 。

    推荐阅读