问题:
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版本不适配
|