在 GCP TPU 上训练我的模型答案

【问题标题】：Train my model on GCP TPU在 GCP TPU 上训练我的模型
【发布时间】：2018-09-25 21:20:34
【问题描述】：

什么架构的程序可以在TPU中运行？

一定要用tf.contrib.tpu.TPUEstimator吗？

除了tensorflow官方模型，有没有tpu-program例子可以参考？

【问题讨论】：

标签： tensorflow google-cloud-platform google-cloud-tpu

【解决方案1】：

Google Cloud TPU 使用支持 Python、C++、Java、Go 和 Swift API [2] 的 TensorFlow [1]。这是 Google Cloud TPU 系统架构 [1] 的链接。 TPUEstimator 处理底层的、特定于硬件的细节，它可以通过代表您自动应用性能优化来更容易地实现最大性能 [3]。此外，如果您使用的是神经网络，那么您应该使用 TPUEstimator。以下是从 TensorFlow Estimator 转换为 TPUEstimator [4] 的一些步骤。其他教程位于 Google Cloud TPU 教程页面 [5] 和 GitHub [6] 上。

[1]https://cloud.google.com/tpu/docs/system-architecture

[2]https://www.tensorflow.org/api_docs/

[3]https://cloud.google.com/tpu/docs/using-estimator-api

[4]https://cloud.google.com/tpu/docs/using-estimator-api#converting_from_tensorflow_estimator_to_tpuestimator

[5]https://cloud.google.com/tpu/docs/tutorials

[6]https://github.com/tensorflow/tpu

【讨论】：