Question

我想将Colab连接到付费TPU（从免费TPU升级）。我使用以下指南创建了一个JSON密钥：https://cloud.google.com/docs/authentication/production#auth-cloud-explicit-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

因此，您似乎正在尝试通过Colab笔记本从自己的Google Cloud项目连接到付费TPU，是吗？由于Colab运行时由与您自己的My-project-name不在同一个项目中的GCE VM支持，因此无法使用。因此，您想在同一项目中创建一个GCE VM，并从该VM运行您的训练脚本。查看本教程：https://cloud.google.com/tpu/docs/quickstart。

将Colab连接到付费TPU

1 个答案: