PyTorch中MNIST数据集(附datasets.MNIST离线包)下载慢/安装慢的解决方案

一、问题背景

在学习MNIST数据集手写数字识别demo的时候,笔者碰到了一些问题,现记录如下:

1、必须先确保torchvision已经正确安。如何安装torchvision?请参考PyTorch/torchvision.datasets自带常用数据集【归纳】

二、解决方案

如果有梯子,直接挂梯子,没有可以参考下面的方案。

1、下载离线安装包

链接: https://pan.baidu.com/s/1pPowtYrT7ETcDQI4o7s9Kw 提取码: 5qfm 复制这段内容后打开百度网盘手机App,操作更方便哦

2、Linux使用unzip mnist-master.zip命令解压,win和Mac请自便。在这里,我把它解压到了file:///home/ftimes/Downloads/mnist.

3、替换原下载路径

  1. 到安装torchvison的目录,搜索minst.py
  2. 打开文件,找到如下的URL
    元路径
  3. 如上图注释掉,改为
    在这里插入图片描述
    也就是前面部分替换为你的本地路径。

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的解决方案

发布了44 篇原创文章 · 获赞 12 · 访问量 9102

猜你喜欢

转载自blog.csdn.net/ftimes/article/details/104620689