I found that the performance of GPU is not good as I expected (as slow as a turtle), I wanna switch from GPU to CPU. but mlcompute module cannot be found, so wired.
The same code ran on colab and my computer (jupyter lab) take 156s vs 40 minutes per epoch, respectively.
I only used a small dataset (a few thousands of data points), and each epoch only have 20 baches.
I am so disappointing and it seems like the "powerful" GPU is a joke.
I am using 12.0.1 macOS and the version of tensorflow-macos is 2.6.0
Can anyone tell me why this happens?
I have observed the same behavior like you, as I mention in the post https://developer.apple.com/forums/thread/709142. I am using tensorflow-macos 2.8.0, tensorflow-metal 0.5.0 and python 3.8.13.