python - 属性错误 : module 'torch.utils' has no attribute '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")