Neh*_*uja 4 python computer-vision importerror pytorch torchvision
即使安装了 pytorch 后,这一行也会出现此错误。
from torchvision import datasets
Run Code Online (Sandbox Code Playgroud)
如果您使用的是 anaconda 发行版,请首先使用以下命令安装 torchvision:
$ conda install -c conda-forge torchvision
Run Code Online (Sandbox Code Playgroud)
如果未安装该软件包,则会安装该软件包。否则,它会抛出消息
# 所有请求的包都已安装。
之后,尝试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',
....,
....
]
Run Code Online (Sandbox Code Playgroud)
正如您从上面的 列表中看到的dir(datasets)
,列出了 的数据集类MNIST
,当正确安装 torchvision 包时就会出现这种情况。