I intalled on my pc tensorflow 2.4.1 and try to test it. I found some simple script at github that train model on mnist:
\'\'\'Trains a simple convnet on the MNI