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

转载已于 2022-05-15 17:32:00 修改 · 1.4w 阅读

39 ·

CC 4.0 BY-SA版权

原文链接：https://blog.csdn.net/qq_28666313/article/details/106343611

文章标签：

#pytorch #深度学习 #python

于 2022-05-08 15:08:25 首次发布

PyTorch 专栏收录该内容

3 篇文章

订阅专栏

本文详细指导如何修复Tensorboard在PyTorch中导入SummaryWriter时的错误，涉及包安装、环境配置，以及从tensorboardX导入的替代方案。适合新手解决常见问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

有时，我们在使用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