安装TensorBoard

在命令行输入

pip install tensorboard

或者在pycharm中安装

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)

运行后会多一个log文件夹（没有的话刷新一下）

在命令行输入

tensorboard --logdir=logs
#tensorboard --logdir=logs --port=6007
# 可以通过--port选项选择端口

点击链接就可以进入了