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)