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
Tested in my Macbook Air m1, it will speed up around 3 times.
Top comments (0)