Gui Castelão

Personal website

May 08, 2023

TensorFlow on M2

I've been having problems running TensorFlow with OSX with M2. I couldn't track yet where is (are) the problem(s), but it seems like everything is changing fast since the M2 release. At first, I couldn't even use GPU. Anyways, this is what has been working for me, and maybe useful for someone out there:

mamba install -c apple tensorflow-deps
pip install tensorflow-macos
pip install tensorflow-metal