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

编程问答 2022-08-16

问题描述

我试图在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 @R_610_4045@ion.
>>> 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'
>>>

解决方法

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

tensorboard

（或类似a）。完成此操作后，您应该从torch.utils包中导入from torch.utils import tensorboard tensorboard.SummaryWriter("foo") 模块：

SummaryWriter

或者您可以直接导入from torch.utils.tensorboard import SummaryWriter SummaryWriter("bar") ：

{{1}}
deep-learningpythonpytorchtensorboard