我正在運行 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
2 回答

皈依舞
TA貢獻1851條經驗 獲得超3個贊
原來這樣做的方法是像這樣直接從 python 啟動進程(我還必須將參數從--tpu
to修改--service_addr
):
import subprocess subprocess.Popen(["capture_tpu_profile","--logdir=gs://<my_logdir>", "--service_addr={}".format(os.environ['COLAB_TPU_ADDR'])])
check=True
如果失敗,則使命令引發異常。
添加回答
舉報
0/150
提交
取消