Same problem here – try running this code for example: https://github.com/mlflow/mlflow/tree/master/examples/tensorflow/tf2 307 .
System details: Win 10, TF 2.6.0, Python 3.8.8.
Please resolve this annoying issue which blocks my project.