arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2108.10470cs.ROcs.LG
收藏

Isaac Gym:用于机器人学习的基于GPU的高性能物理仿真

Isaac Gym: High Performance GPU-Based Physics Simulation For Robot Learning

NVIDIA(英伟达)

Viktor Makoviychuk, Lukasz Wawrzyniak, Yunrong Guo, Michelle Lu, Kier Storey, Miles Macklin, David Hoeller, Nikita Rudin, Arthur Allshire, Ankur Handa, Gavriel State

摘要:

Isaac Gym提供了一个高性能的学习平台,可以直接在GPU上为各种机器人任务进行训练策略。物理模拟和神经网络策略训练都驻留在GPU上,并通过直接将数据从物理缓冲区传递到PyTorch张量进行通信,而无需通过任何CPU瓶颈。这导致在单个GPU上进行复杂机器人任务的快速训练时间,与使用基于CPU的模拟器和用于神经网络的GPU的传统RL训练相比,提高了2-3个数量级。我们在\url{https://sites.google.com/view/isaacgym-nvidia}和isaac健身房托管结果和视频可以在\url{https://developer.nvidia.com/isaac-gym}下载。

英文摘要:

Isaac Gym offers a high performance learning platform to train policies for wide variety of robotics tasks directly on GPU. Both physics simulation and the neural network policy training reside on GPU and communicate by directly passing data from physics buffers to PyTorch tensors without ever going through any CPU bottlenecks. This leads to blazing fast training times for complex robotics tasks on a single GPU with 2-3 orders of magnitude improvements compared to conventional RL training that uses a CPU based simulator and GPU for neural networks. We host the results and videos at \url{https://sites.google.com/view/isaacgym-nvidia} and isaac gym can be downloaded at \url{https://developer.nvidia.com/isaac-gym}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑