Accelerating TensorFlow using Apple M1 Max?

[8bitmp3]

@mrdbourke

I did a bunch of testing across Google Colab, Apple’s M1 Pro and M1 Max as well as a TITAN RTX GPU.

@mrdbourke

Here was one of the experiments using transfer learning with an EfficientNetB0 from tf.keras.applications and the Food101 dataset (~100,000 images) from TensorFlow Datasets, batch size 32.

That’s very interesting, thank you @mrdbourke

@mrdbourke

• Code used for tests - GitHub - mrdbourke/m1-machine-learning-test: Code for testing various M1 Chip benchmarks with TensorFlow.

Thanks for sharing!

---

If anyone is interested, found some older comparisons, though they aren’t about the Max or Pro models:

Benchmark M1 vs Xeon vs Core i5 vs K80 and T4

Benchmark M1 (part 2) vs 20 cores Xeon vs AMD EPYC, 16 and 32 cores