tensorboardX を使って機械学習のlossのグラフがみたい

Question

pythorchでtensorboardを使いたいと思ってます。

https://qiita.com/meipuru344/items/f7ce91627ced8a8a7587

上記サイトを参考にtensorboardを使ってみたのですが、エラーが出力されてしまいます。調べても理由がわからなかったので教えてもらえるとうれしいです。

###やりたいこと
tensorboardXにグラフを書かせてそれをみたい

###環境

• 外部サーバにssh接続している
• NVIDIA Cuda compiler driver ver9.9
• Ubuntu 16.04
• torch 1.2.0
• tensorflow 2.1.0
• tensorboardX 2.0

実行したこと

下のファイルを実行

import tensorboardX as tbx
import torch

#出力先のファイル名指定
writer = tbx.SummaryWriter("runs/exp-1")

for i in range(20):
    writer.add_scalar("group/epoch", torch.randn(1).item(), i)
writer.close()

下のコマンドを実行

tensorboard --logdir="runs/exp-1"

コマンド実行後エラーが出力

2020-02-05 01:59:10.103766: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libnvinfer.so.6'; dlerror: libnvinfer.so.6: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /home/n.nakamura/ubuntu16/lib:/usr/local/cuda/lib64
2020-02-05 01:59:10.105144: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libnvinfer_plugin.so.6'; dlerror: libnvinfer_plugin.so.6: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /home/n.nakamura/ubuntu16/lib:/usr/local/cuda/lib64
2020-02-05 01:59:10.105182: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:30] Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
Traceback (most recent call last):
  File "/home/n.nakamura/ubuntu16/pyenv/versions/3.6.8/bin/tensorboard", line 8, in <module>
    sys.exit(run_main())
  File "/home/n.nakamura/ubuntu16/pyenv/versions/3.6.8/lib/python3.6/site-packages/tensorboard/main.py", line 59, in run_main
    default.get_plugins() + default.get_dynamic_plugins(),
  File "/home/n.nakamura/ubuntu16/pyenv/versions/3.6.8/lib/python3.6/site-packages/tensorboard/default.py", line 115, in get_dynamic_plugins
    for entry_point in pkg_resources.iter_entry_points('tensorboard_plugins')
  File "/home/n.nakamura/ubuntu16/pyenv/versions/3.6.8/lib/python3.6/site-packages/tensorboard/default.py", line 115, in <listcomp>
    for entry_point in pkg_resources.iter_entry_points('tensorboard_plugins')
  File "/home/n.nakamura/ubuntu16/pyenv/versions/3.6.8/lib/python3.6/site-packages/pkg_resources/__init__.py", line 2345, in load
    self.require(*args, **kwargs)
  File "/home/n.nakamura/ubuntu16/pyenv/versions/3.6.8/lib/python3.6/site-packages/pkg_resources/__init__.py", line 2368, in require
    items = working_set.resolve(reqs, env, installer, extras=self.extras)
  File "/home/n.nakamura/ubuntu16/pyenv/versions/3.6.8/lib/python3.6/site-packages/pkg_resources/__init__.py", line 789, in resolve
    raise VersionConflict(dist, req).with_context(dependent_req)
pkg_resources.VersionConflict: (setuptools 40.6.2 (/home/n.nakamura/ubuntu16/pyenv/versions/3.6.8/lib/python3.6/site-packages), Requirement.parse('setuptools>=41.0.0'))

ちなみにssh接続ではなく、ローカルで同じことを実行したときは問題なくやりたいことをすることができました。

Answer 1

エラー文だけで判断しますと、setuptools が古いことで引っかかっているように見えます。
下記コマンドで pip と setuptools を upgrade すると動作する可能性があります。

$ pip install --upgrade pip setuptools

それでも駄目だった場合、念の為 pip freeze で torch や tensorboard がローカルと同じバージョンになっているかも確認すると良いかもしれません。

Answer 2

自己解決したのですが、書き忘れていました。
回答いただいたように、バージョンの問題だったようです。

いろいろ解決方法はあったように思いますが、自分の環境で一番楽に実行できそうなtensorboard のダウングレードで解決しました。
バージョンは2.1.0 -> 1.14　にしました。