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

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐