train_dataset = torchvision.datasets.MNIST(root='./data',
train=True,
transform=transforms.ToTensor(),
download=True)
把mnist数据集连同文件夹一起放到./data下即可