Error AttributeError: module 'tensorflow.compat.v2.__internal__' has no attribute register_load_context_function

[Raghav_Gupta]

Hello

I have just learned deep learning models. So I tried to install the Tensorflow as per the instructions in the link:- Install TensorFlow 2

After installing, when I am doing import keras. System is giving me following errors

import keras
Traceback (most recent call last):
File “”, line 1, in
File “C:\Users\ragha\AppData\Local\Programs\Python\Python39\lib\site-packages\keras_init_.py”, line 21, in
from keras import models
File “C:\Users\ragha\AppData\Local\Programs\Python\Python39\lib\site-packages\keras\models_init_.py”, line 18, in
from keras.engine.functional import Functional
File “C:\Users\ragha\AppData\Local\Programs\Python\Python39\lib\site-packages\keras\engine\functional.py”, line 34, in
from keras.engine import training as training_lib
File “C:\Users\ragha\AppData\Local\Programs\Python\Python39\lib\site-packages\keras\engine\training.py”, line 45, in
from keras.saving.legacy import save
File “C:\Users\ragha\AppData\Local\Programs\Python\Python39\lib\site-packages\keras\saving\legacy\save.py”, line 24, in
from keras.saving.legacy.saved_model import load_context
File “C:\Users\ragha\AppData\Local\Programs\Python\Python39\lib\site-packages\keras\saving\legacy\saved_model\load_context.py”, line 68, in
tf.internal.register_load_context_function(in_load_context)
AttributeError: module ‘tensorflow.compat.v2.internal’ has no attribute ‘register_load_context_function’

So I need help in resolving this error. So that I can work on Tensor Flow models.

[mtalhaubaid]

can you share you code here? and which verison of keras and tensorflow you are using

[chunduriv]

To use Keras, you will need to have the TensorFlow package installed. See detailed instructions.

Once TensorFlow is installed, just import Keras via:

from tensorflow import keras

Please try as suggested above and let us know if the problem continues? Thank you

[Raghav_Gupta]

Thanks for your help.

I have removed the tf conda environment completely and then installed the environment again.

I have then again installed all the libraries as mentioned in the detailed instructions link.

But after installing, I was again getting some different errors of module not found for keras. So I then installed tensorflow in the main environment too (Without conda tf environment). Post this I was able to run the tensorflow code

[arefyazdi4]

I found that problem is from Keras packages
so I execute these commands

pip uninstall keras
pip install tensorflow