在GCP TPU上训练我的模型

Question

TPU可以运行哪种程序架构？

是否必须使用tf.contrib.tpu.TPUEstimator？

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

Answer 1

Google Cloud TPU使用TensorFlow [1]，它支持Python，C ++，Java，Go和Swift API [2]。以下是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