一、MNIST数据集下载

可以从这里下载MNIST数据集。
下载解压缩后，可以看到我们要用的四个文件：

train开头的是训练数据，分为images和labels。
t10k开头的是测试数据，分为images和labels。
如果想要训练集、验证集和测试集，可以自己读取数据后划分。

二、MNIST数据集使用

1.直接调用torchvision.datasets.MNIST函数读取。
需要将下载后的MNIST数据按照一定路径存放，这里提供一个可以读取的方式：
①在项目文件夹下新建MNIST文件夹，在MNIST文件夹下新建raw文件夹，将数据存放在raw文件夹下。数据路径如下：

②python文件中使用torchvision.datasets.MNIST函数读取数据：

import torch
import torchvision
import torchvision.transforms as transforms

trainset = torchvision.datasets.MNIST(root='./', train=True, download=False, transform=transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=batch_size_train)