谷歌开放第二/三代云端TPU Pod 可配备1000颗TPU

OPEN编辑 5年前

       北京时间 5 月 8 日早间消息,据美国科技媒体 TechCrunch 报道,谷歌今天宣布其第二代和第三代 Cloud TPU Pod 将以 beta 版的形式提供给公众。

  这两款基于云计算的超级计算机最多将配置 1000 颗定制 TPU。最新的 v3 型号性能尤其强大,而且配有液冷系统。每一款设备都可以提供高达每秒 100 千万亿次浮点数的运算能力。

  谷歌表示,按照原始运算能力计算,这种设备可以跻身全球前五大超级计算机行列,但由于 TPU 的数值精度很低,所以在考虑这个数值时应该打个折扣。

  但用户不必非要使用完整的 TPU Pod,谷歌还允许开发者租赁其中的部分算力。但无论如何,这款设备都非常强大,最快可以在两分钟时间内训练一个使用 ImageNet 数据集的标准 ResNet-50 图片分类模型。

  TPU v2 Pod 最多配备 512 个核心,运算速度略慢于 v3。例如,当使用 265 个 TPU 时,v2 Pod 可以在 11.3 分钟内训练 ResNet-50 模型,v3Pod 仅需 7.1 分钟。使用一个 TPU 花费的时间为 302 分钟。(鼎宏)