Pytorch 导入tersorboard 的 SummaryWriter时报错

Pytorch 导入tersorboard 的 SummaryWriter时报错,第1张

问题概述:

Pytorch 导入tersorboard 的 SummaryWriter时报错:raise ImportError(‘TensorBoard logging requires TensorBoard version 1.15 or above’)
ImportError: TensorBoard logging requires TensorBoard version 1.15 or above
或者
from torch.utils.tensorboard import SummaryWriter
ImportError: cannot import name ‘SummaryWriter’ from ‘torch.utils.tensorboard’


问题描述
   writer = SummaryWriter("logs")
    tensor_trans = transforms.ToTensor()
    tensor_img = tensor_trans(img)
    trans_resize = transforms.Resize((224, 224))
    # PIL->(resize)->PIL
    img_resize = trans_resize(tensor_img)
    img_resize=img_resize.unsqueeze(0)
    print(tensor_img.shape)
    # input = torch.ones([1,1,224,224])
    output=Su(img_resize)
    writer.add_images("randomcorp", output, 1)
    writer.close()


原因分析:

例如:具体原因可能是因为自己的版本问题 或者 在自己的项目下存在同名(tensorboard.py)文件


解决方案:

版本问题:

     pip install tensorboard=1.15.0 #1.15.0是版本 一般14.0以上都可以

同名问题:

     #删除或修改同名py文件

欢迎分享,转载请注明来源:内存溢出

原文地址:https://54852.com/langs/798869.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-05-06
下一篇2022-05-06

发表评论

登录后才能评论

评论列表(0条)

    保存