pytorch下载minist

PyTorch是一个流行的深度学习框架，如果你想在PyTorch中下载MNIST数据集，可以按照以下步骤操作：

1. **安装必要的库**：首先，确保已经安装了`torchvision`，它包含了常用的图像处理数据集。如果没有安装，你可以通过pip进行安装：

   pip install torchvision

2. **导入模块并加载数据**：使用`torchvision.datasets.MNIST`直接下载并加载MNIST数据：

   import torch
   from torchvision import datasets, transforms

   # 数据预处理
   transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.5,), (0.5,))])

   # 加载训练集和测试集
   train_dataset = datasets.MNIST(root='./data', train=True, download=True, transform=transform)
   test_dataset = datasets.MNIST(root='./data', train=False, download=True, transform=transform)

这里会自动下载MNIST的数据到`./data`目录下。

3. **查看数据**：

   for images, labels in train_dataset:
       break
   print(images.shape, labels.shape)  # 输出：(64, 1, 28, 28) (64,)