基于PyTorch的深度学习6——可视化工具Tensorboard

Wis4e2025-03-11 20:43

先安装tensorflow（CPU或GPU版），然后安装tensorboardX，在命令行运行以下命令即可。

pip install tensorboardX

使用tensorboardX的一般步骤如下所示。

1)导入tensorboardX，实例化SummaryWriter类，指明记录日志路径等信息。

复制代码

from tensorboardX import SummaryWriter
#实例化SummaryWriter，并指明日志存放路径。在当前目录没有logs目录将自动创建。
writer = SummaryWriter(log_dir='logs')
#调用实例
writer.add_xxx()
#关闭writer
writer.close()

2)调用相应的API接口，接口一般格式为：

复制代码

add_xxx(tag-name, object, iteration-number)
#即add_xxx（标签，记录的对象，迭代次数）

3)启动tensorboard服务：cd到logs目录所在的同级目录，在命令行输入如下命令，logdir等式右边可以是相对路径或绝对路径。

复制代码

tensorboard --logdir=logs --port 6006   
#如果是Windows环境，要注意路径解析，如
#tensorboard --logdir=r'D:\myboard\test\logs' --port 6006

4)web展示。在浏览器输入：

复制代码

http://服务器IP或名称:6006  #如果是本机，服务器名称可以使用localhost