在ARM 版本的 Mac M1 上安裝 TensorFlow

Yungger
4 min readOct 4, 2021

STEP 1 : 安裝 Python 開發環境 (Conda)

可直接參考我的另一篇文章: 在ARM 版本的 Mac M1 上安裝 Python (Mini-forge Conda 虛擬環境)

STEP 2: 建置專案的虛擬環境

可直接參考我的另一篇文章: 在ARM 版本的 Mac M1 上安裝 Python (Mini-forge Conda 虛擬環境)

須注意的是 Apple tensorflow_macos 目前支援的最新版本只到 3.8。

所以, 建立虛擬環境時的參數, 建議如下:

  • python=3.8 指定版本為 3.8, 因為 Apple tensorflow_macos 目前支援的最新版本只到 3.8。
  • 參數 p, 表示安裝至指定路徑
  • 參數 y, 表示安裝過程的須對話輸入時, 自動回答 yes

*** 以下假設已依上述文章步驟, 建立了一個專案 tensorflow 的虛擬環境 ***

STEP 3 : 啟動 (進入) 虛擬環境

% conda activate tensorflow

進入專案的虛擬環境後, 可先檢查一下環境資訊與Python版本是否正確?

  • conda info
  • python -V

STEP 4 : 安裝相關所需的套件

conda install -y astunparse
conda install -y six
conda install -y numpy
conda install -y absl-py
conda install -y google-pasta
conda install -y google-api-python-client
conda install -y gast
conda install -y termcolor
conda install -y python-flatbuffers
conda install -y wrapt
conda install -y opt_einsum
conda install -y tensorboard

STEP 5 : 安裝 TensorFlow 與 Addons for macOS

至 [Mac-optimized TensorFlow and TensorFlow Addons] (https://github.com/apple/tensorflow_macos/releases) 點選後, 要取得的是:

然後, 在專案的虛擬環境下的終端機, 輸入

pip install --upgrade --force --no-dependencies https://github.com/apple/tensorflow_macos/releases/download/v0.1alpha3/tensorflow_addons_macos-0.1a3-cp38-cp38-macosx_11_0_arm64.whl  https://github.com/apple/tensorflow_macos/releases/download/v0.1alpha3/tensorflow_macos-0.1a3-cp38-cp38-macosx_11_0_arm64.whl

最後, 測試一下, 結果顯示已可以正常在 M1 機器上的使用 Tensorflow 了。

接下來, 專案的環境一旦建立後, 您就可以順利的在上面執行寫好的 TensorFlow 程式了 !!

作者:Yungger 勇哥

若覺得此文真對您有幫助, 點點左上方的拍手圖示, 鼓勵一下。若還想 請喝杯咖啡 鼓勵一下勇哥的繼續創作, 也感謝 😘 😘 !!

Yungger

Maker & Freelancer

--

--