docker上のjupyterlabでtensorboardの表示

docker上でやると多少コマンドやオプション等が異なったので残しておく

magic commandを用いることでjupyter notebookのセル上でtensorboardを表示することができる。これをdocker上のjupyterlabにて表示させてみる

環境

• python: 3.7
• jupyterlab: 1.2.4
• tensorflow: 2.2.0
• tensorboard: 2.2.1

手順

1. tensorboard用にdockerのportsを設定しておく

version: '3'
services:
  app:
    build: .
    volumes:
      - ./:/app
    ports:
      - "8888:8888" # for jupyter
      - "6006:6006" # for tensorboard
    command: jupyter lab --ip=0.0.0.0 --allow-root --NotebookApp.token='' --port=8888

2. notebook内でtensorboardを表示するにはmagic commandを使えばいい

この時hostオプションを設定しておく また、./modelsの下に、ログを保存している

%load_ext tensorboard

%tensorboard --logdir ./models --host 0.0.0.0

流れとしては↓のようになる

参考

• Using TensorBoard in Notebooks  |  TensorFlow