利用pytorch加載mnist數據集的代碼如下

import torchvision
import torchvision.transforms as transforms
from torch.utils.data import DataLoader

train_data = torchvision.datasets.MNIST(
    root=‘./mnist/‘,
    train=True,                                     # this is training data
    transform=torchvision.transforms.ToTensor(),    #
 
 Converts a PIL.Image or numpy.ndarray to
                                                    # torch.FloatTensor of shape (C x H x W) and normalize in the range [0.0, 1.0]
    download=True,
)
test_data = torchvision.datasets.MNIST(
    root=‘./mnist/‘,
    train=False,                                     #
 
 this is training data
    transform=torchvision.transforms.ToTensor(),    # Converts a PIL.Image or numpy.ndarray to
                                                    # torch.FloatTensor of shape (C x H x W) and normalize in the range [0.0, 1.0]
    download=True,
)

train_data_loader = DataLoader(train_data, shuffle=True, batch_size = 100)
test_data_loader 
 
= DataLoader(test_data,shuffle=True, batch_size=100)

第一次使用mnist，需要下載，具體方法就是設置download=True，然而我運行的時候報錯了。

錯誤：not gzip file

可是明明是.gz文件啊，查了幾篇博客也沒有說清楚原因的，於是自行下載了四個文件（訓練集，測試集以及各自的標簽），放入./mnist/raw文件夾下，運行，報錯：找不到文件。

此時，系統需要找的是./mnist/process文件夾下的train.pt和test.pt文件，這應該是pytorch下載原文件後處理生成的，可是我無處下載，於是用另一臺電腦下載，程序沒有報錯，我把生成的.pt文件拷貝過來，可以運行了

pytorch 加載mnist數據集報錯not gzip file