python - ImportError：没有名为“torchvision.datasets.mnist”的模块

Question

即使在安装 pytorch 之后，此行也会出现此错误。

from torchvision import datasets

标签： pythoncomputer-visionpytorchimporterrortorchvision

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 包的情况。