【pytorch】pytorch 1.2中tensorboard无法显示graph解决办法

背景

这两天使用 pytorch 1.2.0 版本自带的tensorboard, 却发现无法显示出模型的graph,无论是官网的教程还是网上的示例还是自己写的例子,均不成功.出现情况如下:

我的环境: Python3.7 pytorch 1.2.0 torchvision 0.4.0 tensorboard 1.14 在上网查了pytorch forum 和其他论坛发现很多人出现了和我一样的情况.

解决办法 1. 替换为 tensorboardX

1. 根据安装方法卸载原有tensorboard
   pip 安装

   pip uninstall tensorboard
   

   anaconda 安装

   conda uninstall tensorboard
   

2. 安装 tensorboardX

   pip install tensorboardX
   

3. 在脚本中替换为tensorboardX
   把如下引用

   from torch.utils.tensorboard import SummaryWriter
   

   替换为

   from tensorboardX import SummaryWriter
   

解决办法 2. 升级 pytorch到 1.3.0 级以上

• 升级到 preview 版的pytorch-nightly即可
  以anaconda为例:

  conda install pytorch torchvision cudatoolkit=10.0 -c pytorch-nightly
  

  (我选择的cudatoolkit为10,可以根据具体情况去掉或者更改)
  其他安装方法详见官网

结果