导入错误：没有名为“torchvision.datasets.mnist”的模块

Question

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

from torchvision import datasets

Answer 1

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

$ conda install -c conda-forge torchvision

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

# 所有请求的软件包已安装。

此后，尝试导入您提到的

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 软件包时就会出现这种情况。

Answer 2

mnist 必须大写

torchvision.datasets.MNIST