- tensorboardX是pytorch的可视化工具
- tensorboard作为Tensorflow中强大的可视化工具,已经被广泛使用
但针对其他框架,例如Pytorch,之前一直没有这么好的可视化工具可用,好在目前Pytorch也可以支持Tensorboard了,那就是通过使用tensorboardX,真是Pytorcher的福利! - 因为tensorboardX是对tensorboard进行了封装后,开放出来使用,所以必须先安装tensorboard, 再安装tensorboardX!!
必须先安装tensorboard再安装tensorboardX!!!否则会报错 - 安装步骤
step1 安装tensorflow,安装tensorflow会自动安装tensorboard
可以参考另外一篇文章安装tensorflow
pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
step2 确定tensorboard安装好之后安装tensorboardX
pip install tensorboardX
step3 tensorboardX已经全部安装好,在Terminal中输入tensorboard --logdir=路径,点击链接即可
参考文献:pytorch可视化tensorboard