You need to use the tensorflow-metal version 0.5.0. See the version table on https://developer.apple.com/metal/tensorflow-plugin/. Install the proper version with: python -m pip install tensorflow-metal==0.5.0