我可以将 TensorBoard 与 Google Colab 一起使用吗?

2023-12-07

在Google Colab上训练TensorFlow模型时有没有办法使用TensorBoard?


EDIT:您可能想给官方%tensorboard magic尝试一下,从 TensorFlow 1.13 开始可用。


在存在之前%tensorboard魔法,标准方法 实现这一目标的方法是使用以下命令将网络流量代理到 Colab VMngrok。可以找到 Colab 示例here.

这些是步骤(代码片段代表 colab 中“代码”类型的单元格):

  1. 让 TensorBoard 在后台运行。
    灵感来自这个答案.

    LOG_DIR = '/tmp/log'
    get_ipython().system_raw(
        'tensorboard --logdir {} --host 0.0.0.0 --port 6006 &'
        .format(LOG_DIR)
    )
    
  2. 下载并解压ngrok.
    替换传递给的链接wget以及适合您操作系统的正确下载链接。

    ! wget https://bin.equinox.io/c/4VmDzA7iaHb/ngrok-stable-linux-amd64.zip
    ! unzip ngrok-stable-linux-amd64.zip
    
  3. 启动 ngrok 后台进程...

    get_ipython().system_raw('./ngrok http 6006 &')
    

...并检索公共网址。Source

    ! curl -s http://localhost:4040/api/tunnels | python3 -c \
        "import sys, json; print(json.load(sys.stdin)['tunnels'][0]['public_url'])"
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

我可以将 TensorBoard 与 Google Colab 一起使用吗? 的相关文章

随机推荐