一、问题背景
在学习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

- 如上图注释掉,改为

本文提供了一种在无法直接访问网络的情况下,解决MNIST数据集加载问题的方法。通过下载离线安装包并手动替换torchvision中的数据下载链接,使PyTorch能够从本地读取数据,避免了网络限制带来的困扰。
最低0.47元/天 解锁文章
1019





