apple_m1_pro_python

A collection of ML scripts to test the M1 Pro MacBook Pro

Stars
166
Committers
2

Apple Silicon DL benchmarks

Currently we have PyTorch and Tensorflow that have Metal backend.

Results

Varied results across frameworks:

Tensorflow Resnet50:

PyTorch Resnet50:

  • Difference between CPU and GPU
    gpu_vs_cpu.png
  • Comparing with Nvidia
    samples_sec.png

PyTorch Bert

  • Running a Bert from Huggingface
    pt_bert.png

Pytorch

We have official PyTorch support! check pytorch folder to start running your benchmarks

Tensorflow

You can run tensorflow benchmarks by going to the tensorflow folder.