🤩 TensorbordX

壹丨安装¶

pip install tensorboardX

pip install tensorboard

使用GPU的情况下，需要安装：

pip install tensorflow-gpu==2.3.0

注意：版本需要和cuda版本一致¹

否则报cudart64_110.dll not found错误，可根据参考[1]查找对应版本

import tensorflow as tf
print(tf.__version__)

贰丨使用¶

在runs同级目录下执行：

tensorboard --logdir runs

叁丨总结¶

用法详见参考²³，个人觉得不好用，PyTorch训练可视化可以试试visdom 。