Instalace Tensorflow na Mac OS Apple M1

Při zakládání projektu v pyCharm se vyberte virtuální prostředí o dané verzi Pythonu.

v terminálu se bude postupovat následovně.

Požadavky

• Počítač Mac s čipem Apple nebo GPU AMD
• macOS 12.0 nebo novější
• Python 3.8 nebo novější
• Nástroje příkazového řádku Xcode: xcode-select --install

Nastavení

Stáhněte si prostředí Conda

bash miniconda.sh -b -p miniconda
source miniconda/bin/activate
conda install -c apple tensorflow-deps

Nainstalujte základnu TensorFlow

python -m pip install tensorflow-macos

Nainstalujte modul tensorflow-metal

python -m pip install tensorflow-metal

Otestujte

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)

Při testu může dojít k vypsání zprávy, že nebyl nalezen GPU, ale je to jenom chybná hláška protože v monitoru aktivit je vidět, že GPU je vytěžované.