【M1 Mac】Tensorflowのインストール方法

強力なApple Siliconを使って深層学習をしてみたいと、MacBook Pro M1 Proを購入しました。早速、深層学習フレームワークの一つ、TensorFlowをインストールしてみましたので、備忘として残しておきたいと思います。

すでに仮想環境が設定され、pythonがインストールされていることが前提となります。

目次

TensorFlowのインストール

Appleのホームページにインストール方法が記載されていますので、それに従えば簡単にインストールすることができます。

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

この3つのコマンドを順番に打ち込むだけでインストールが完了します。

あとは、pythonを起動して、

>>>import tensor flow

と入力して、エラーが出なければOKです。

目次