Error: ModuleNotFoundError: No module named 'tensorflow.contrib'

Question

I'm using Tensorflow 2.0 but getting error while executing train.py.

from tensorflow.contrib.memory_stats.python.ops.memory_stats_ops import BytesInUse ModuleNotFoundError: No module named 'tensorflow.contrib'

can anyone please help me with finding tf.contrib module in tensorflow2.0

Answer 1

tensorflow.contrib is being removed in version 2.0, you therefore need version <= 1.14 to operate tflearn (by the way, this is a TFlearn issue, not a tensorflow one). ( source )