Подключите Colab к платному ТПУ

Question

Я бы хотел подключить Colab к платному TPU (обновление с бесплатного TPU). Я создал ключ JSON, используя это руководство: https://cloud.google.com/docs/authentication/production#auth -cloud-явный- python, а затем загрузил его в Colab. Я могу подключиться к своему хранилищу, но не к TPU:

%tensorflow_version 2.x
import tensorflow as tf
import os
os.environ['GOOGLE_APPLICATION_CREDENTIALS'] = './gcp-permissions.json'

# Authenticated API request - works.
storage_client = storage.Client.from_service_account_json(
    'gcp-permissions.json')
print(list(storage_client.list_buckets())

#Accessing the TPU - does not work. Request times out.
cluster_resolver = tf.distribute.cluster_resolver.TPUClusterResolver(
    tpu='My-TPU-Name',
    zone='us-central1-a',
    project='My-Project-Name'
)

Я также попробовал вызов TPUClusterResolver только с именем tpu и с 'credentials = gcp-permissions. json '- тот же результат. Я дважды проверил, что мой TPU работает и работает в консоли GCP. Это не выкупаемо. Чего мне не хватает?

Спасибо!

Answer 1

Похоже, вы пытаетесь подключиться к платному TPU из собственного проекта Google Cloud из ноутбука Colab, верно? Это не сработает, поскольку среда исполнения Colab поддерживается виртуальной машиной GCE, которая находится в другом проекте, чем ваш My-project-name. Таким образом, вместо этого вы также хотите создать виртуальную машину GCE в этом же проекте и запустить сценарий обучения с этой виртуальной машины. Оформить заказ: https://cloud.google.com/tpu/docs/quickstart.