【Pytorch/工具】Pytorch中TensorBoard的使用

TensorBoard的安装及使用

安装

【工具】安装TensorBoard

pip install tensorboard

输入上述命令，回车等待安装即可

使用

使用该函数生成日志文件

from torch.utils.tensorboard import SummaryWriter
writer=SummaryWriter("logs")
# writer.add_image()
for i in range(100):
    writer.add_scalar("y=x",i,i)
writer.close()

运行完上面的代码之后，可以看到在改项目下面生成了一个log文件夹

查看日志文件：

tensorboard --logdir=logs

默认使用的是6006端口
可以手动指定使用的端口：

tensorboard --logdir=logs --port=6009

在浏览器中输入网址或者点击网址自动打开浏览器该网址:

http://localhost:6009/

这里需要注意一下，有的同学可能出现下图所示的问题：
意思是没有写入任何事件文件！！！
可能是您的terminal所在的路径不对，一定要将目录进入到logs文件夹的上一层

比如我这里是在gan项目下面建立了test_code然后运行刚刚写入的事件的代码，它生成的logs文件夹默认跟py文件在同一层目录下，因为项目根目录 是gan，所以terminal打开默认路径也是gan，此时就要cd test_code才可以运行tensorboard --logdir=logs --port=6009该命令。