gpt4 book ai didi

google-cloud-platform - 将 Colab 连接到付费 TPU

转载 作者:行者123 更新时间:2023-12-04 19:32:06 25 4
gpt4 key购买 nike

我想将 Colab 连接到付费 TPU(从免费 TPU 升级)。我使用本指南创建了一个 JSON key :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 控制台中启动并运行。它不是可抢占的。我错过了什么?

谢谢!

最佳答案

因此,您似乎正在尝试通过 Colab 笔记本连接到您自己的 Google Cloud 项目中的付费 TPU,对吗?这将不起作用,因为 Colab 运行时由与您自己的项目不同的 GCE VM 支持 My-project-name .因此,您还希望在同一个项目中创建一个 GCE VM,并从该 VM 运行您的训练脚本。查看本教程:https://cloud.google.com/tpu/docs/quickstart .

关于google-cloud-platform - 将 Colab 连接到付费 TPU,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59796410/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com