如何在 Google 云中 select 特定的 TPU？

Question

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

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

但它没有说明如何 select 一个 TPU，它只给出了 select 一个区域的指令。

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

我想知道如何 select 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

您可以使用 tpu-size 参数 select TPU 类型，根据 the documentation (also here)。

例如：

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

请记住，只有 v2-8 和 v3-8 are available 除非您可以获得评估配额或已购买承诺。

Answer 2

您还可以使用 gcloud 命令创建 TPU

gcloud compute tpus create tpu-quickstart \
      --zone=us-central1-a \
      --network=default \
      --accelerator-type=v2-32 \
      --version=2.1