加载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)

保存退出。

重新运行,问题解决。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐