在Google Colab上使用TPU错误_无法在系统中找到任何TPU内核

Question

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

什么是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 /副本：0 / task：0 / device：CPU：0，CPU，   268435456、13626261043432226130），   _DeviceAttributes（/ job：localhost /副本：0 / task：0 / device：XLA_CPU：0，XLA_CPU，17179869184、240953179956275984）]。