tensorflow - 在 Google Colab 上使用 TPU Error_Cannot find any TPU cores in the system

Question

我正在尝试使用 TPU 在 Google Colab 上运行BERT 模型。

TPU_NAME 到底是什么：

!python run_classifier.py  \
.
.
.
--use_tpu=True \
--tpu_name=$TPU_NAME

我试图将其定义为：

 TPU_NAME = os.environ['COLAB_TPU_ADDR']

但是，我遇到了以下错误：

INFO：tensorflow：从 training_loop 记录的错误：在系统中找不到任何 TPU 核心。请仔细检查 TensorFlow 主地址和 TPU 工作人员。可用设备为 [_DeviceAttributes(/job:localhost/replica:0/task:0/device:CPU:0, CPU, 268435456, 13626261043432226130), _DeviceAttributes(/job:localhost/replica:0/task:0/device:XLA_CPU :0, XLA_CPU, 17179869184, 240953179956275984)]。