Pytorch学习笔记之from torch.utils.tensorboard import SummaryWriter导入不成功问题

有时，我们在使用tensorboard时，会遇到形如“ImportError: cannot import name 'SummaryWriter' from partially initialized module 'torch.utils.tensorboard' ”的问题。

由于pytorch本身不自带tensorboard包，一般这样报错都是由于未安装tensorboard包导致的。

可通过Anaconda中的Anaconda Prompt进入安装有pytorch的虚拟环境，执行以下命令：

pip install tensorboard

正确安装tensorboard之后，重新运行代码，若还是报错在torch.utils.tensorboard中找不到SummaryWriter。则再继续下面的步骤。

因为SummaryWriter是存在于tensorboardX(其作为tensorboard的子模块)因此同样通过pip进行安装：

pip install  -i https://mirrors.aliyun.com/pypi/simple/ tensorboardX

这里是通过清华镜像源进行安装的。

如果正确安装tensorboardX依旧找不到SummaryWriter
请将报错的：
       

from torch.utils.tensorboard import SummaryWriter

改为：    
       

from tensorboardX import SummaryWriter

————————————————
版权声明：本文为CSDN博主「向日葵炒瓜子」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/qq_28666313/article/details/106343611