运行 Google Colab 环境中的云 TPU 分析器

Question

我正在运行使用 Google Colab 笔记本并尝试捕获 TPU 分析数据以用于 TensorBoard，但是我无法获得 capture_tpu_profile 到运行在后台运行运行我的 TensorFlow 代码。

到目前为止，我尝试运行在后台捕获进程：

!capture_tpu_profile --logdir=gs://<my_logdir> --tpu=$COLAB_TPU_ADDR &

和

!bg capture_tpu_profile --logdir=gs://<my_logdir> --tpu=$COLAB_TPU_ADDR

Answer 1

原来一个方法是像这样直接从 python 开始进程（我还必须将参数从 --tpu 修改为 --service_addr）：

import subprocess
subprocess.Popen(["capture_tpu_profile","--logdir=gs://<my_logdir>", "--service_addr={}".format(os.environ['COLAB_TPU_ADDR'])])

check=True 使命令在失败时引发异常。

Answer 2

一种方法是使用 TPUProfilerHook

它将探查器作为会话挂钩运行。

Running cloud TPU profiler in Google Colab environment