加载数据集报错The given NumPy array is not writeable, and PyTorch does not support non-writeable tensors.

加载MNIST数据集报错The given NumPy array is not writeable, and PyTorch does not support non-writeable tensors.

解决办法:

找到torchvision下dataset目录,以文本方式打开mnist数据集

例如:E:\Anaconda3\envs\python37\Lib\site-packages\torchvision\datasets下

打开文件,找到return torch.from_numpy(parsed.astype(m[2], copy=False)).view(*s)

将copy=False删除,变为:

return torch.from_numpy(parsed.astype(m[2])).view(*s)

保存退出。

重新运行,问题解决。

你可能感兴趣的:(pytorch,深度学习,机器学习)