[TensorFlow] No module named 'tensorflow.examples.tutorials'

코랩에서 MNIST 실습을 하려던 중 보게 된 에러 메시지이다. No module named 에러는 pip(또는 !pip) install로 웬만하면 다 해결되는데, pip install tensorflow-examples로 해결되지 않아서 구글링해봤다.

 

출처는 https://github.com/tensorflow/tensorflow/issues/32790 이고 내가 사용한 방법은 1)이다.

 

import 사용

import tensorflow_datasets as tfds

# Construct a tf.data.Dataset
dataset = tfds.load(name="mnist", split=tfds.Split.TRAIN)

 

tensorflow_datasets이라는 모듈을 import하고 name에서 mnist임을 명시해주면 아래와 같이 아주 깔끔하게 실행된다.

 

 

import 사용 X

TF 2.X에서는 특정 모듈을 import하지 않아도 tf.keras.datasets.mnist 만으로 mnist 데이터 사용이 가능하다고 한다.