Google ColaboratoryのTPUの利用で環境変数'COLAB_TPU_ADDR'がなくてエラー

Question

Google ColabのTPUを利用してローカル接続でCNNを作成中です。
「ノートブックの設定」でハードウェアアクセラレータは「TPU」を選択し、「接続済み(ローカル)」となっています。
しかし、ホスト型ランタイムでは読み込んでアドレスが返ってくるのですが、ローカル接続では環境変数がNoneで返ってくるためgRPCから進めず困っています。

発生している問題・エラーメッセージ

None

---------------------------------------------------------------------------

KeyError                                  Traceback (most recent call last)

<ipython-input-6-022a85c6413d> in <module>
      4 print(os.environ.get("COLAB_TPU_ADDR"))
      5 
----> 6 tpu_grpc_url = "grpc://" + os.environ["COLAB_TPU_ADDR"]
      7 tpu_cluster_resolver = tf.contrib.cluster_resolver.TPUClusterResolver(tpu_grpc_url)
      8 tf.contrib.distribute.initialize_tpu_system(tpu_cluster_resolver)

~/anaconda3/lib/python3.7/os.py in __getitem__(self, key)
    676         except KeyError:
    677             # raise KeyError with the original key value
--> 678             raise KeyError(key) from None
    679         return self.decodevalue(value)
    680 

KeyError: 'COLAB_TPU_ADDR'

該当のソースコード

import tensorflow as tf
import os

print(os.environ.get("COLAB_TPU_ADDR"))

tpu_grpc_url = "grpc://" + os.environ["COLAB_TPU_ADDR"]
tpu_cluster_resolver = tf.contrib.cluster_resolver.TPUClusterResolver(tpu_grpc_url)
tf.contrib.distribute.initialize_tpu_system(tpu_cluster_resolver)
strategy = tf.contrib.distribute.TPUStrategy(tpu_cluster_resolver, steps_per_run=100)

試したこと

Tensorflowを1.4と1.13.1でも変わらず。
Python 3.6.9でも変わらず。
ファイアウォールを無効にしてみても変わらず。

補足情報（FW/ツールのバージョンなど）

anacondaで管理
Python 3.7.3
Tensorflow 1.13.1
Firefox 68.0.2

Answer 1

ローカルということは、自分でTPUなどの環境を用意する必要がありますが、それは出来ていますか？

Comments

[kazuma_kishi]

失礼致しました。
ローカルで自前のTPUの環境ではないのでGPUで構築してみます。