If you followed "https://developer.apple.com/metal/tensorflow-plugin/" to install TF, they you should be already taking advantage of the GPU power. The fact that your TF can see your GPU (device Metal) also confirmed that. How slow is your training process? For your reference, training the CIFAR-100 example in the above link takes 90s per epoch on my M1 Pro GPU. In my benchmarks, M1 Pro GPU is 5 - 7 times faster than M1 Pro CPU and should be around the level of an RTX 3050 Mobile for deep learning. It is not going to be as fast as a good desktop GPU.
Topic:
Graphics & Games
SubTopic:
General
Tags: