M1 OS 12.4 に Miniconda で Pytorch と Tensorflow をインストールする

概要

Apple M1 Pro, Macbook Pro で、Anaconda をアンイストールし、Miniconda をインストール、

  • Pytorch
  • Tensorflow

をインストールする。

特に Tensorflow のインストールでトラブルが多いため様々な情報があるが、私の環境でのインストールを書きます。

tensorflow はパッケージの バージョンの組み合わせによっては動作しないパターンが有る ということ（要約するとこれが今回の記事のポイントです）。

ご参考まで。

環境

  • Macbook Pro 14 inch
  • Apple M1 Pro
  • OS 12.4 (Monterey)
  • Anaconda インストール済み

今まで特に Deep Learning を使わずに分析や機械学習等やってきましたが、画像を扱う案件があり、データセキュリティ上ローカルでの構築が必要で環境を作り直しました。

Anaconda のアンイストール

$ conda install -c anaconda anaconda-clean
$ anaconda-clean
Delete .conda? (y/n): y
Delete .ipython? (y/n): y
Delete .jupyter? (y/n): y
...

$ rm -rf ~/opt/anaconda3

参考にしたサイト qiita.com

Miniconda のインストール

Pytorch のサイトを参考にした。

$ curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-arm64.sh
$ sh Miniconda3-latest-MacOSX-arm64.sh

pytorch.org

Miniconda では環境を作ったときに python さえもインストールされないミニマル環境なのでビビりました。

$ conda create -n my_new_env

Pytorch のインストール

上記の Pytorch サイトを参照し

$ conda install pytorch torchvision -c pytorch

$ ipython 
In [2]: import torch
In [3]: x = torch.rand(5, 3)
In [4]: print(x)

Tensorflow のインストール

トラブル多し。ポイントは、

  • tensorflow-deps
  • tensorflow-macos
  • tensorflow-metal

のバージョンの組み合わせによっては動作しないパターンが有るということ（これが今回の記事のポイントです）。

developer.apple.com

私の環境では次のバージョンの組み合わせで動作した。

$ conda install -c apple tensorflow-deps==2.10.0
$ python -m pip install tensorflow-macos==2.10.0 tensorflow-metal==0.6.0

$ ipython
In [2]: import tensorflow as tf
In [3]: tf.__version__
Out[3]: '2.10.0'

OS 13.1 の人も上記フォーラムでは

tensorflow-deps 2.10, tensorflow-macos 2.10 and tensorflow-metal 0.6.0

の組み合わせでうまく行ったとのこと。

一方、OS 12.4 では下記の組み合わせでもうまくいったという報告があります。

tensorflow-deps 2.9, tensorflow-macos 2.9 and tensorflow-metal 0.6.0