Я пытаюсь использовать Google Colab для использования Tensor Processing Unit (TPU) для обучения нейронной сети. Tensorflow только что выпустил основной выпуск 2.0, поэтому я пытаюсь сделать это в Tensorflow 2.0. Я попробовал следующие три руководства, но все они были написаны для Tensorflow 1.14 - и терпят неудачу с Tensorflow 2.0:
1) Следуя руководству ТПУ в Colab , я получаюошибка:
AttributeError: module 'tensorflow' has no attribute 'Session'
(из ссылки: с tf.Session (tpu_address) в качестве сеанса: )
2) Послеруководство Простая классификационная модель с использованием Keras на Colab TPU , я получаю ту же ошибку
3) Следуя руководству cloud_tpu_custom_training , я получаю ошибку:
AttributeError: module 'tensorflow' has no attribute 'contrib'
(из ссылки: resolver = tf.contrib.cluster_resolver.TPUClusterResolver (tpu = TPU_WORKER) )
У кого-нибудь есть пример использованияTPU для обучения нейронной сети в Tensorflow 2.0?
Редактировать: Эта проблема также возникает на github: InvalidArgumentError: Невозможно найти context_id, соответствующий указанному # 1