Mac M1/M2にGPUサポート付きのTensorFlowをインストール

Mac M1/M2でGPUサポートを備えたTensorFlowを数ステップでインストールし、新しいMac Silicon ARM64アーキテクチャのネイティブパフォーマンスを活用する。Mac M1/M2が際立っているのは、卓越した性能だけでなく、極めて低い電力消費にもある。

Xcodeコマンドラインツールをインストール

もしそれがすでにシステムにインストールされていない場合、Mac OSXのターミナルで以下のコマンドを実行してインストールできます：

xcode-select --install

M1/M2バージョンのAnacondaをインストール

Anaconda はデータサイエンス向けの環境を提供するプラットフォームです。科学技術計算などを中心とした、多くのモジュールやツールのコンパイル済みバイナリファイルを提供しており、簡単にPythonを利用する環境を構築できます。

Anaconda はPythonだけではなく、いろいろなユーティリティや他のプログラミング言語・ライブラリなども、パッケージ管理ツール Conda でインストールできます。curlなどの便利なユーティリティや、NvidiaのGPUを利用する場合に必要なCUDAなどの環境もインストールできるようになっています。

Anacondaのダウンロード：https://www.anaconda.com/download

TensorFlowをインストール

TensorFlowの依存関係をインストール:

conda install -c apple tensorflow-deps

基本のTensorFlowをインストール

pip install tensorflow-macos

Metal プラグインをインストール（tensorflow-metalがなければ、 M1/M2 GPUが利用できない）

pip install tensorflow-metal

GPUの利用可能性を確認

import sys

import tensorflow.keras
import pandas as pd
import sklearn as sk
import scipy as sp
import tensorflow as tf
import platform

print(f"Python Platform: {platform.platform()}")
print(f"Tensor Flow Version: {tf.__version__}")
print(f"Keras Version: {tensorflow.keras.__version__}")
print()
print(f"Python {sys.version}")
print(f"Pandas {pd.__version__}")
print(f"Scikit-Learn {sk.__version__}")
print(f"SciPy {sp.__version__}")
gpu = len(tf.config.list_physical_devices('GPU'))>0
print("GPU is", "available" if gpu else "NOT AVAILABLE")

Python Platform: macOS-13.0-arm64-arm-64bit
Tensor Flow Version: 2.11.0
Keras Version: 2.11.0

Python 3.9.16 (main, Jan 11 2023, 10:02:19)
[Clang 14.0.6 ]
Pandas 1.5.3
Scikit-Learn 1.2.1
SciPy 1.10.0
GPU is available