ImportError:没有名为“ torchvision.datasets.mnist”的模块

时间:2019-01-20 08:22:51

标签: python computer-vision pytorch importerror torchvision

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

from torchvision import datasets

1 个答案:

答案 0 :(得分: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软件包时的情况。