M1 MacとPoetryでTensorFlow環境構築

はじめに

Apple Silicon 搭載 Mac にTensorFlow 入り環境 を miniforge ではなく Poetry を使って構築する際に、ちょっと苦労したので備忘録です。

こちらの記事を参考にさせていただきました。

前提

• Homebrew がインストールされている
• Python のパッケージマネージャは Poetry を使用している

インストールに成功したモジュールは以下のバージョンのものです。

poetry.lock

[[package]]
name = "tensorflow-macos"
version = "2.5.0"

[[package]]
name = "tensorflow-metal"
version = "0.1.2"

手順

1. Homebrew で必要なものを事前にインストール

shell

brew install hdf5 zlib openblas openssl

2. 環境変数付きで tensorflow-macos を poetry add

当方の環境では、参考記事 に記載があった環境変数に加え、openblas と openssl についても環境変数を明示するとうまくいきました。

shell

CFLAGS="-I /opt/homebrew/opt/openssl/include" LDFLAGS="-L /opt/homebrew/opt/openssl/lib" GRPC_PYTHON_BUILD_SYSTEM_OPENSSL=1 GRPC_PYTHON_BUILD_SYSTEM_ZLIB=1 HDF5_DIR=/opt/homebrew/opt/hdf5 OPENBLAS="/opt/homebrew/opt/openblas" poetry add tensorflow-macos

3. tensorflow-metal を poetry add

shell

poetry add tensorflow-metal

さいごに

良い tensorflow-metal ライフを！