模型训练可视化工具——TensorboardX

安装

$ pip install tensorboardX
$ pip install tensorboard
# 使用GPU的情况下，需要安装
$ pip install tensorflow-gpu==2.3.0

版本需要和cuda版本一致，否则报cudart64_110.dll not found错误

https://blog.csdn.net/weixin_42764932/article/details/113038416

# 查看TensorFlow版本：
$ python
$ import tensorflow as tf
$ tf.__version__

使用

在与runs同级目录下执行：

$ tensorboard --logdir runs

传送门

https://www.jianshu.com/p/46eb3004beca

https://blog.csdn.net/bigbennyguo/article/details/87956434