tensorflow - 在 GCP TPU 上训练我的模型

Question

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

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

除了 tensorflow 官方模型，有没有任何 tpu-program 示例可供参考？

Answer 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