一、TensorBoard
1、安装
在PyTotch中使用TensorBoard是,首先要确保安装了以下包:
pip install tb-nightly
pip install future
输入from torch.utils.tensorboard import SummaryWriter
后不会出错,则表明环境安装正确
- 若没有安装
tb-nightly
,则输入from torch.utils.tensorboard import SummaryWriter
后,会出现如下错误:
2.若没有安装future
,则输入from torch.utils.tensorboard import SummaryWriter
后,会出现如下错误:
2、使用
1.tensorboard --logdir=runs
使用上述命令时,可能会出现如下错误: