Как перейти с Colab TPU на платную TPU? - PullRequest
Новая
       13

Как перейти с Colab TPU на платную TPU?

0 голосов
/ 14 января 2020

За последние несколько дней я использовал Colab с TPU, и это было здорово. Теперь я хотел бы перейти на платный TPU на GCE. Я следил за быстрым стартом здесь: https://cloud.google.com/tpu/docs/quickstart

Я создал корзину хранения и TPU, используя tpu up. Теперь хотел бы подключиться к TPU с помощью моего ноутбука, размещенного на GCP AI Notebooks (https://console.cloud.google.com/ai-platform/notebooks). Когда я был на Colab, я мог получить к нему доступ с помощью: 

cluster_resolver = tf.distribute.cluster_resolver.TPUClusterResolver()
tf.config.experimental_connect_to_cluster(cluster_resolver)

Когда я запускаю ту же строку в GCP AI Notebook ... 

cluster_resolver = tf.distribute.cluster_resolver.TPUClusterResolver('MyTPUName')
tf.config.experimental_connect_to_cluster(cluster_resolver)

... он выдает это ошибка 

HttpError: <HttpError 403 when requesting https://tpu.googleapis.com/v1/projects/local-dialect-[MyProjectID]/locations/us-east1-c/nodes/[MyTPUName]?alt=json returned "Location us-east1-c is not found or access is unauthorized.">

Как это исправить? Кстати, если кто-то из Google читает это: если вы собираетесь брать с этого продукта 1-5 долларов в час, как насчет кнопки «обновить» от Colab? Все побеждают.

Спасибо!

1 Ответ

0 голосов
/ 16 января 2020

Оказывается, мне нужно было несколько дополнительных параметров, чтобы пройти через безопасность. Это сработало: 

cluster_resolver = tf.distribute.cluster_resolver.TPUClusterResolver( \
        tpu='[my tpu name]', 
        zone='us-central1-a', 
        project='[my project name]')
tf.config.experimental_connect_to_cluster(cluster_resolver)
...