pytorch1.7.1安装tensorboard

0.打开终端

conda install tensorboard

尽管存在torch.utils.tensorboard ，但是pytorch内部并没有安装包，需要自己安装

1.代码示例

代码里和tensorboard相关的都写的了注释
SummaryWriter()自动创建一个runs的文件夹

里面存了一个event的文件

import torch
from torch.utils.tensorboard import SummaryWriter


# Writer will output to' ./runs/' directory by default.
writer = SummaryWriter()

x = torch.arange(-5, 5, 0.1).view(-1, 1)
y = -5 * x + 0.1 * torch.randn(x.size())

model = torch.nn.Linear(1, 1)
criterion = torch.nn.MSELoss()
optimizer = torch.optim.SGD(model.parameters(), lr=0.1)


def train_model(iter):
    for epoch in range(iter):
        y1 = model(x)
        loss = criterion(y1, y)
        # log a scalar value，使用add_scalar函数来完成
        # For example, lets create a simple linear regression training,
        # and log loss value using add_scalar
        # 参数含义：title，y轴，x轴
        writer.add_scalar("Loss/train", loss, epoch)
        optimizer.zero_grad()
        loss.backward()
        optimizer.step()


train_model(10)
writer.flush()
writer.close()
# 在终端使用tensorboard logdir= ‘路径名’
# 把runs的文件路径给出来就行，tensorborad会自动递归的寻找带有event后缀的文件
# 然后直接打开 http://localhost:6006/

2.参考来源

pytorch文档之tensorboard