Post

Replies

Boosts

Views

Activity

TensorFlow slower using GPU then using CPU with M1 Pro
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.
3
0
2.2k
Dec ’21