ImportError: нет модуля с именем «torchvision.datasets.mnist»

Question

Даже после установки pytorch эта ошибка появляется для этой строки.

from torchvision import datasets

Answer 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.