使用 torch.utils.tensorboard 看不到图表答案

【问题标题】：Can't see graph using torch.utils.tensorboard使用 torch.utils.tensorboard 看不到图表
【发布时间】：2020-02-07 23:46:50
【问题描述】：

我正在尝试习惯 tensorboard，并使用 pytorch 对模型进行编码。但是，当我尝试使用 add_graph() 函数查看我的模型时，我得到了这个：

以此作为测试代码：

import numpy as np
import torch
import torchvision.transforms as transforms
import torch.nn as nn
import torch.optim as optim
from torch.utils.tensorboard import SummaryWriter

class Net(nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        self.linear = nn.Linear(2, 1)

    def forward(self, x):
        x = self.linear(x)
        return x

writer = SummaryWriter('runs_pytorch/test')
net = Net()
criterion = nn.CrossEntropyLoss()
optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)
writer.add_graph(net, torch.zeros([4, 2], dtype=torch.float))
writer.close()

另一方面，如果我尝试使用 TensorFlow 查看图表，一切似乎都很好：

这次用这个作为测试代码：

import tensorflow as tf
tf.Variable(42, name='foo')
w = tf.summary.FileWriter('runs_tensorflow/test')
w.add_graph(tf.get_default_graph())
w.flush()
w.close()

如果你想知道，我正在使用这个命令来启动 tensorboard：

tensorboard --logdir runs_pytorch

我注意到，当我在为我的 tensorflow 测试分配的目录上使用它时，我得到了带有地址的通常消息，但是如果我用 --logdir runs_pytorch 做同样的事情，我还有更多的东西：

W1010 15:19:24.225109 15308 plugin_event_accumulator.py:294] 每次运行发现多个图事件，或者存在包含 graph_def 的元图以及一个或多个图事件。用最新事件覆盖图表。 W1010 15:19:24.226075 15308 plugin_event_accumulator.py:322] 发现多个带有标签 step1 的“运行元数据”事件。用最新的事件覆盖它。

我在 Windows 上，我尝试过不同的浏览器（chrome、firefox...）。我有 tensorflow 1.14.0、torch 1.2.0、Python 3.7.3

非常感谢您的帮助，这让我发疯了！

【问题讨论】：

标签： tensorflow graph pytorch tensorboard

【解决方案1】：

有两种方法可以解决：

1.将 PyTorch 更新到 1.3.0 及更高版本：

康达方式：

    conda install pytorch torchvision cudatoolkit=9.2 -c pytorch

点子方式：

    pip3 install torch==1.3.0+cu92 torchvision==0.4.1+cu92 -f https://download.pytorch.org/whl/torch_stable.html

2.改为安装 tensorboardX：

卸载张量板：

如果你的张量板是通过 pip 安装的：
```
    pip uninstall tensorboard
```
如果你的张量板是由 anaconda 安装的：
```
    conda uninstall tensorboard
```
安装 tensorboardX
```
    pip install tensorboardX
```

在编写脚本时，改变

    from torch.utils.tensorboard import SummaryWriter

到

    from tensorboardX import SummaryWriter

【讨论】：

【解决方案2】：

这可能是由 known problem 引起的，并且似乎已在昨天发布的 pytorch 1.3 中解决 - 查看 release notes 中的错误修复。

【讨论】：