安装依赖:
sudo pip install tensorboardX
sudo pip install tensorboard
sudo pip install tensorflow
如果在安装tensorflow过程中出现cant uninstall enum34错误
可以先运行以下命令
sudo pip install --ignore-installed enum34
然后重新安装
sudo pip install tensorflow
scalar使用:
from tensorboardX import SummaryWriter
writer = SummaryWriter(log_dir='xxx')
创建一个summarywriter,指定记录文件的保存目录为xxx
writer.add_scalars('aaa/bbb', x, i)
x代表想要记录的标量(纵坐标),i代表迭代次数(横坐标)
cd到xxx上一级目录,命令行输入:
tensorboard --logdir xxx
打开输出结果中的链接,就可以在网页上看到可视化结果