即使已安装了pytorch,该行仍会出现此错误。
from torchvision import datasets
from torchvision import datasets
conda install 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包正确安装时的情况。torchvision.datasets.MNIST
torchvision
?可以通过执行命令pip3 install torchvision
进行安装。 - kHarshitpip install --upgrade --force-reinstall torchvision
重新安装torchvision。 - Monster