DEV Community

BC
BC

Posted on

Use macbook m1 GPU to train tensorflow model

By default when you train model with model.fit, it will still use your CPU (Macbook m1), you can let it use GPU train with an extra installation:

python -m pip install tensorflow-macos python -m pip install tensorflow-metal 
Enter fullscreen mode Exit fullscreen mode

Tested in my Macbook Air m1, it will speed up around 3 times.

Top comments (0)