Hello,
I have recently bought the M1 Pro and installed Tensorflow using the indications of Apple.
I noted that training a simple NN with GPU was really slow (~30 secs per epoch). However, after forcing Tensorflow to use CPU I saw an incredible improvement (~3 secs per epoch).
I cannot believe that the difference is that much. Verified with the System Monitor and I saw Tensorflow using 100% of the GPU.
Could anyone assist me or enlighten me?
Please check below the code used to train this model.