如何从 Colab TPU 升级到付费 TPU？

Question

过去几天，我一直在使用带有 TPU 的 Colab，效果非常好。现在，我想升级到 GCE 上的付费 TPU。我一直在关注这里的快速入门：https://cloud.google.com/tpu/docs/quickstart

我已经使用 tpu up 创建了一个存储桶和一个 TPU。现在想使用我的笔记本连接到 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 的 "upgrade" 按钮怎么样？大家都赢了。

谢谢！

Answer 1

原来我需要一些额外的参数才能通过安全检查。这有效：

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)

如何从 Colab TPU 升级到付费 TPU？

How to upgrade from Colab TPU to paid TPU?

jupyter-notebook

tpu

gcp-ai-platform-notebook