Question

过去几天，我一直在将Colab和TPU一起使用，这真是太好了。现在，我想在GCE上升级到付费的TPU。我一直在这里关注快速入门：https://cloud.google.com/tpu/docs/quickstart

我已经使用tpu up创建了一个存储桶和一个TPU。现在，我想使用由GCP AI笔记本（https://console.cloud.google.com/ai-platform/notebooks）托管的笔记本连接到TPU。当我在Colab上时，可以使用以下命令进行访问：

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

当我在GCP AI笔记本中运行同一行时...

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的“升级”按钮怎么样？每个人都赢。

谢谢！

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？

1 个答案: