重磅 | 谷歌首次透露 TPU 细节：处理速度是 GPU/CPU 的 15-30 倍

雷峰网 · 公众号 · 科技媒体 · 2017-04-06 12:16

正文

请到「今天看啥」查看全文

TensorFlow 机器学习框架之外，就再未透露更多的细节。今日，谷歌的硬件工程师 Norm Jouppi 首次向外分享了更多关于该项目的细节和测试结果。

如果你是一个芯片设计师，你可以在谷歌公布的研究报告里找到很多关于这一 TPU 如何运作的细节。

在此次测试中，谷歌基于自己的基准测试对自研芯片进行了测试。测试结果表明，TPU 在执行谷歌常规机器学习工作负载方面，比一个标准的 GPU/CPU 组合（一般是 Intel Haswell 处理器和 Nvidia K80 GPU 组合的情况下）平均要快 15-30 倍。另外，由于数据中心的功耗计算，TPU 还能提供高达 30-80 倍瓦特的速率。研究报告作者表示，如果将来使用更快的内存，该 TPU 还有进一步优化的空间。

值得注意的是，这些数字是关于在生产中使用机器学习模型的，而不是首次创建模型。

谷歌还指出，虽然大多数架构师为卷积神经网络（convolutional neural networks，例如，对于图像识别工作良好的特定类型的神经网络）优化了其芯片。然而，谷歌表示，这些网络只占其数据中心工作负载的 5％左右，而大部分应用使用的是多层感知器（ multi-layer perceptrons）。