导入错误:找不到名为'torchvision.datasets.mnist'的模块。

4
即使已安装了pytorch,该行仍会出现此错误。
from torchvision import datasets

请确认一下,您是否也安装了 torchvision?可以通过执行命令 pip3 install torchvision 进行安装。 - kHarshit
尝试使用 pip install --upgrade --force-reinstall torchvision 重新安装torchvision。 - Monster
2个回答

6
如果您正在使用Anaconda发行版,请先使用以下方式安装torchvision:

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包正确安装时的情况。

3
对于除了Anaconda以外的任何其他发行版,PyTorch主页提供了一个方便的工具,可以让你输入你的环境,并返回一个命令来安装相关的包,如果你向下滚动一点的话。 - dennlinger

0
mnist必须大写

torchvision.datasets.MNIST


请投票关闭拼写错误的问题,而不是回答它们。您可以在评论中解释拼写错误。不鼓励回答应该关闭的问题(请参见[答案])。 - undefined
@starball 另一个回答者甚至没有解决问题,这应该更加令人沮丧。请停止发表无用的评论。 - undefined

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接