Error in tensorflow loading model valueerror: unknown layer: functional

Contents of articles

Problem description solution references

Problem description

When you pull the model trained by the server to your own computer, tensorflow loads the model and reports an error valueerror: unknown layer: functional error

import tensorflow as tf

model = tf.keras.models.load_model('test.h5')

The server

python 3.6.13
tensorflow-gpu==2.3.0

Own computer

python 3.6.5
tensorflow-gpu==2.1.0

Solution

pip install tensorflow-gpu==2.3.0

References

    Value