Question

我正在运行Google Colab笔记本，并试图捕获TensorBoard中使用的TPU分析数据，但是在运行TensorFlow代码时，我无法让capture_tpu_profile在后台运行。

到目前为止，我尝试使用以下命令在后台运行捕获过程：

!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

哪个配置文件作为会话挂接器运行。