一、问题背景
在学习MNIST数据集手写数字识别demo的时候,笔者碰到了一些问题,现记录如下:
1、必须先确保torchvision已经正确安。如何安装torchvision?请参考PyTorch/torchvision.datasets自带常用数据集【归纳】
二、解决方案
如果有梯子,直接挂梯子,没有可以参考下面的方案。
1、下载离线安装包
【2020年4月18日更新】:点击蓝奏云下载链接 MNIST数据集(https://ww.lanzous.com/ibjwnoj)
【失效】链接: https://***/s/1pPowtYrT7ETcDQI4o7s9Kw 提取码: 5qfm
2、Linux使用unzip mnist-master.zip命令解压,win和Mac请自便。在这里,我把它解压到了file:///home/ftimes/Downloads/mnist.
3、替换原下载路径
- 到安装torchvison的目录,搜索minst.py
- 打开文件,找到如下的URL
- 如上图注释掉,改为
也就是前面部分替换为你的本地路径。
4.重载后,成功运行。
5.如果提示EOFError: Compressed file ended before the end-of-stream marker was reached,
请参考处理PyTorch中EOFError: Compressed file ended before the end-of-stream marker was reached的解决方案