在 google colab 中，有没有办法检查 TPU verison 是什么运行？

In google colab, is there a way to check what TPU verison is running?

colab 提供免费的 TPU。很容易看出给定了多少个内核，但我想知道是否可以查看每个内核有多少内存？

据我所知，我们没有用于访问内存信息的 Tensorflow 操作或类似操作，但在 XRT 中我们有。与此同时，类似以下代码片段的内容是否有效？

import os
from tensorflow.python.profiler import profiler_client

tpu_profile_service_address = os.environ['COLAB_TPU_ADDR'].replace('8470', '8466')
print(profiler_client.monitor(tpu_profile_service_address, 100, 2))

输出如下：

  Timestamp: 22:23:03
  TPU type: TPU v2
  Utilization of TPU Matrix Units (higher is better): 0.000%

TPUv2 每核 8GB，TPUv3 每核 16GB HBM (https://cloud.google.com/tpu)。