【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,可以根据具体情况去掉或者更改)
    其他安装方法详见官网

结果

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 技术黑板 设计师:CSDN官方博客 返回首页