Google Cloud ML 提供 TPU 定制芯片答案

【问题标题】：TPU custom chip available with Google Cloud MLGoogle Cloud ML 提供 TPU 定制芯片
【发布时间】：2017-02-24 09:04:48
【问题描述】：

在使用 TensorFlow 时，将哪种类型的硬件用作 Google Cloud ML 的一部分？也只有 CPU 或张量处理单元（自定义卡）可用？

【问题讨论】：

【解决方案1】：

Cloud ML 目前专注于 CPU。 GPU 和 TPU 将在未来推出。

【讨论】：

【解决方案2】：

在撰写本文时（2017 年 12 月），GPU 可用，请参阅 https://cloud.google.com/ml-engine/docs/training-overview

如果您使用gcloud 命令行实用程序，您可以例如在向 mlengine 提交作业时添加 --scale-tier BASIC_GPU 选项。这目前可以在 Tesla K80 上运行您的 tensorflow 代码。

还有一个CUSTOM 扩展层，它允许更复杂的配置并提供对 P100 GPU 的访问（请参阅https://cloud.google.com/ml-engine/docs/using-gpus）。

根据https://cloud.google.com/tpu/，TPU 服务处于“alpha”状态，需要注册以了解更多信息。

【讨论】：

【解决方案3】：

Cloud TPU 自 2018 年 6 月 27 日起向公众开放：https://cloud.google.com/tpu/docs/release-notes

这是在 Google Next '18 上宣布的：

【讨论】：