AttributeError：模块“torch.utils”没有属性“tensorboard”

Question

我尝试在 torch.utils 中使用tensorboard，但它说“模块'torch.utils'没有属性'tensorboard'”。我的火炬版本是“1.6.0+cu101”

PS C:\Users\kelekelekle> python
Python 3.7.6 (tags/v3.7.6:43364a7ae0, Dec 19 2019, 01:54:44) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
1.6.0+cu101
>>> writer = torch.utils.tensorboard.SummaryWriter()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: module 'torch.utils' has no attribute 'tensorboard'
>>>

Answer 1

您必须tensorboard通过以下方式安装：

pip install tensorboard

（或类似）。完成后，您应该tensorboard从包中导入模块torch.utils：

from torch.utils import tensorboard

tensorboard.SummaryWriter("foo")

SummaryWriter或者您可以直接导入：

from torch.utils.tensorboard import SummaryWriter

SummaryWriter("bar")