Tensorflow + GPU Acceleration on Apple Silicon Mac

2023.12.17

TECH

環境

M1 Macbook Air

conda 23.11.0

方法

condaの環境作成 (create conda environment)

conda create --name <ENV_NAME> python=3.10

conda環境のアクティベート (activation)

conda activate <ENV_NAME>

Xcode Command Line Toolのインストール (未インストール時) (Install Xcode Command Line Tool)

xcode-select --install

pip のアップグレード (Upgrade pip)

python -m pip install --upgrade pip

tensorflowのインストール(Install tensorlfow)

conda install -c apple tensorflow-deps
python -m pip install tensorflow-macos
python -m pip install tensorflow-metal

追加モジュールのインストール (Install additional modules)

conda install pandas scikit-learn scipy
python -m pip install keras==3

Jupyter

Jupyter Notebookのインストール (Install jupyter notebook)

conda install notebook -y

実行 (Run jupyter)

jupyter lab

Jupyter Notebook実行時にchardet モジュールがないというエラーが出る時 (If chardet not found error happened)

conda install chardet

GPUが使えるか確認する

import sys
import 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: {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-14.2-arm64-arm-64bit
Tensor Flow Version: 2.15.0
Keras Version: 3.0.0

Python 3.10.13 (main, Sep 11 2023, 08:16:02) [Clang 14.0.6 ]
Pandas 2.1.4
Scikit-Learn 1.3.0
SciPy 1.11.4
GPU is available

Homeへ戻る