赞
踩
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版本不适配
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。