强化AI学习 NVIDIA新作将训练速度大幅提升
来源:静雅生活网 作者:仔猪发布时间:2021-09-06 16:58阅读次
导读:英伟达NVIDIA为研究人员开发了Isaac Gym高性能机器人模拟平台。它可以运行一个端到端的GPU加速训练管道,使研究人员能够克服诸多限制,在连续控制任务中
网友提问:
强化AI学习 NVIDIA新作将训练速度大幅提升
优质回答:
英伟达NVIDIA为研究人员开发了Isaac Gym高性能机器人模拟平台。它可以运行一个端到端的GPU加速训练管道,使研究人员能够克服诸多限制,在连续控制任务中实现100倍-1000倍的训练速度。
Isaac Gym通过将物理缓存中的数据直接传递到 PyTorch 张量进行通信,可以在 GPU 上端到端地实现物理模拟和神经网络策略训练。 Isaac Gym 提供了一个高性能的学习平台,可以直接在 GPU 上进行各种代理训练。与使用 CPU 模拟器和 GPU 神经网络的传统 RL 训练相比,Isaac Gym 大大减少了单个 GPU 上复杂机器任务的训练时间,其训练速度提高了 1-2 个数量级。
版权声明:本文部分来自互联网,由小编精心所写,本文地址:http://www.zhubian88.cn/smbk/4302.html,如需转载,请注明出处!