当前位置:   article > 正文

pytorch导包Dataloader失败原因_cannot import name 'dataloader' from 'torch.utils.

cannot import name 'dataloader' from 'torch.utils.data

问题:

pytorch中导入’Dataloader’失败。
报错:
ImportError: cannot import name ‘Dataloader’ from ‘torch.utils.data’ (C:\Users\sq\anaconda3\envs\pytorch-gpu\lib\site-packages\torch\utils\data_init_.py)

解决方案:

把这句:
from torch.utils.data import Dataloader
改成:
from torch.utils.data import _utils

并且将代码中Dataloader改为torch.utils.data.DataLoader

例如:
原代码:
cifar_train = DataLoader(cifar_train,batch_size = batchsz,shuffle=True)
修改为:
cifar_train = torch.utils.data.DataLoader(cifar_train,batch_size = batchsz,shuffle=True)

原因是torch版本不适配


声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop】
推荐阅读
相关标签
  

闽ICP备14008679号