PyTorch可视化工具tensorboard安装

PyTorch可视化工具tensorboard安装

一.安装

tensorboard原本是tensorflow的可视化工具，但自PyTorch1.2.0版本开始，PyTorch正式内置tensorboard的支持，尽管如此仍需手动安装tensorboard。否则会报错。

ModuleNotFoundError: No module named ‘tensorboard’

进入相应虚拟环境后，输入以下指令即可。

pip install tensorboard

二.验证

输入以下指令

from torch.utils.tensorboard import SummaryWriter

不报错即可。