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 установлен правильно.

Answer 2

mnist должно быть написано с заглавной буквы

torchvision.datasets.MNIST