解决pytorch本地导入mnist数据集报错问题

问题描述：

众所周知国内的网络的问题，mnist(MNIST, FashionMNIST, EMNIST)数据集不搭梯子现在是不能自动下载了。

当我去网上下载好mnist数据集之后，据说要放在：

然后我的代码是这样的：

import torch
from torch.utils.data import DataLoader
from torch import nn, optim
from torchvision import transforms, datasets

import visdom


def main():
    mnist_train = datasets.MNIST(root='./mnist', train=True, transform=transforms.Compose([
        transforms.ToTensor()
    ]), download=False)
    mnist_train = DataLoader(mnist_train, batch_size=32, shuffle=True)

    mnist_test = datasets.MNIST(root='./mnist', train=False, transform=transforms.Compose([
        transforms.ToTensor()
    ]), download=False)
    mnist_test = DataLoader(mnist_test, batch_size=32, shuffle=True)
    x, _ = iter(mnist_train).next()
    print('x:', x.shape)


if __name__ == '__main__':
    main()

运行之后是这样的：

Dataset not found. You can use download=True to download it

root='./mnist这个路径改了又改也不行

最后发现是缺少俩文件：

---

解决方法：

github上一个大佬提供了EMNIST全系列的.pt文件

链接：torchvision-datasets-mnist

最后的目录（代码没改，路径还是./mnist）：

运行完美：

如果大佬的链接失效了，那我给出我的（只有mnist）：

链接：https://pan.baidu.com/s/1f7Ok62Gg2xcnz4ZpQiVXiw
提取码：jdzl
复制这段内容后打开百度网盘手机App，操作更方便哦–来自百度网盘超级会员V5的分享