我正在運行 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 啟動進程(我還必須將參數從--tputo修改--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
提交
取消
