kma*_*o23 6

如果您使用的是 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 包时就会出现这种情况。

  • 对于任何其他发行版(不是 Anaconda),[PyTorch 主页](https://pytorch.org/) 提供了一个方便的工具,可以让您输入环境,并返回命令来安装相关包,只需向下滚动一点点。 (3认同)