AI在线 AI在线

WarpDrive

一块V100运行上千个智能体、数千个环境，这个「曲率引擎」框架实现RL百倍提速

一块V100运行上千个智能体、数千个环境，这个「曲率引擎」框架实现RL百倍提速
在强化学习研究中，一个实验就要跑数天或数周，有没有更快的方法？近日，来自 SalesForce 的研究者提出了一种名为 WarpDrive（曲率引擎）的开源框架，它可以在一个 V100 GPU 上并行运行、训练数千个强化学习环境和上千个智能体。实验结果表明，与 CPU+GPU 的 RL 实现相比，WarpDrive 靠一个 GPU 实现的 RL 要快几个数量级。
9/12/2021 12:43:00 PM
机器之心
  • 1