tensorflow - 如何在 Google Cloud 中选择特定的 TPU？

Question

我正在尝试在 Google 云上使用 TPU，并且正在尝试弄清楚如何指定要使用的正确 TPU。我正在尝试遵循快速入门

https://cloud.google.com/tpu/docs/quickstart

但它并没有说如何选择TPU，它只给出了选择区域的说明。

$ ctpu up --zone=us-central1-b  \
--tf-version=2.1 \
--name=tpu-quickstart

我想知道如何选择 v2-32 。起初我想我应该只指定 us-central1-a 但我注意到这里的区域可以容纳不止一种 TPU 类型

https://cloud.google.com/tpu/docs/types-zones

比如us-central1-a既有v2-128也有v2-32，所以我不太确定单凭region就可以指定TPU类型。我有点害怕不小心启动付费 TPU。

Answer 1

根据 the documentation，您可以使用 tpu-size 参数选择 TPU 类型(还有 here )。

例如:

ctpu up --zone=us-central1-a  \
--tf-version=2.1 \
--name=tpu-quickstart \
--tpu-size=v2-32

记住只有 v2-8 和 v3-8 are available除非您有权访问评估配额或已购买 promise 。