即使在安装pytorch之后,此行也会出现此错误。
from torchvision import datasets
答案 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软件包时的情况。