python - 导入错误:没有名为 'torchvision.datasets.mnist' 的模块

Question

即使安装了 pytorch 后，这一行也会出现此错误。

from torchvision import datasets

Answer 1

如果您使用的是 anaconda 发行版，请首先使用以下命令安装 torchvision:

$ conda install -c conda-forge torchvision

如果该软件包尚未安装，则会安装它。否则，它将抛出消息

# All requested packages already installed.

此后，尝试按照您提到的方式导入torchvision.datasets。

In [1]: from torchvision import datasets 

In [2]: dir(datasets)  
Out[2]: 
['CIFAR10',
 'CIFAR100',
 'CocoCaptions',
 'CocoDetection',
 'DatasetFolder',
 'EMNIST',
 'FakeData',
 'FashionMNIST',
 'ImageFolder',
 'LSUN',
 'LSUNClass',
 'MNIST',
 'Omniglot',
 'PhotoTour',
 'SEMEION',
 'STL10',
 'SVHN',
 ....,
 ....
]

正如您从上面的 dir(datasets) 列表中看到的，列出了 MNIST 的数据集类，当正确安装 torchvision 软件包时就会出现这种情况。