python - 在 Google Colab Pro 中使用 TPU v3

Question

有没有办法在 Google Colab Pro 中使用 TPU v3 而不是 TPU v2？

不幸的是，我收到一条错误消息 编译失败:内存空间 hbm 中的内存不足。使用 8.29G 的 7.48G hbm。超过 hbm 容量 825.60M。 使用 TPU v2，我不再收到 TPU v3。因为 TPU v3 具有更多内存。

有人知道可能性/选项吗？

我用这个启动了 TPU

try:
  tpu = tf.distribute.cluster_resolver.TPUClusterResolver()  # TPU detection
  print('Running on TPU ', tpu.cluster_spec().as_dict()['worker'])
except ValueError:
  raise BaseException('ERROR: Not connected to a TPU runtime; please see the previous cell in this notebook for instructions!')


tf.config.experimental_connect_to_cluster(tpu)
tf.tpu.experimental.initialize_tpu_system(tpu)
#tpu_strategy = tf.distribute.experimental.TPUStrategy(tpu)
strategy = tf.distribute.TPUStrategy(tpu)

Answer 1

简短的回答是否定的。无法指定所需的特定 TPU 版本。我相信 Kaggle 提供了 v3-8 TPU(由于它是免费的，因此也可能会发生变化)。此外，正如另一个答案所指出的，您也可以自己启动付费 Cloud TPU，您可以为其指定特定硬件。