How to run TensorFlow on Apple Mac M1, M2 with GPU support

How to run TensorFlow on Apple Mac M1, M2 with GPU support

 

 

1. Xcode command-line tools: xcode-select --install

2. anaconda(miniconda) setup

https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-arm64.sh

bash ~/miniconda.sh -b -p $HOME/miniconda
source ~/miniconda/bin/activate

3. Install tensorflow-deps for apple mac

conda install -c apple tensorflow-deps

4. Install TensorFlow (Base)

python -m pip install tensorflow-macos

5. Install tensorflow-metal plug-in

python -m pip install tensorflow-metal

6. Run tensorflow codes

import tensorflow as tf

cifar = tf.keras.datasets.cifar100
(x_train, y_train), (x_test, y_test) = cifar.load_data()
model = tf.keras.applications.ResNet50(
    include_top=True,
    weights=None,
    input_shape=(32, 32, 3),
    classes=100,)

loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)
model.compile(optimizer="adam", loss=loss_fn, metrics=["accuracy"])
model.fit(x_train, y_train, epochs=5, batch_size=64)