macOS M1(AppleSilicon) 安裝TensorFlow環境

蘋果為M1晶片的Mac提供了TensorFlow的支援，可以使用M1晶片進行硬體加速，以下是如何給使用M1晶片的macOS安裝TensorFlow的環境。

首先，在安裝之前，我們需要先安裝好conda-forge的miniforge3。具體可參考《MacOS M1(Apple Silicon) 安裝配置 Conda 環境》。

建立conda環境

miniforge3預設是python3.9，我們安裝tensorflow需要3.8版本的python，是以我們需要先建立一個conda環境

conda create -—name python38 python=3.8
# 激活環境
conda activate python38
           

下載下傳Apple提供的tensorflow支援

下載下傳位址：https://github.com/apple/tensorflow_macos/releases，選擇tar.gz的包下載下傳。

下載下傳完成後，使用tar進行解壓

tar -xvf tensorflow_macos-0.1alpha2.tar.gz
           

安裝環境前準備工作

接下來，需要先配置兩個變量，一個是剛下載下傳的安裝包位置，另一個是目标安裝環境的位置。如果不清楚具體位置的可以進入目錄使用pwd指令擷取路徑。

# $Home 為自己使用者目錄，根據實際情況自行修改
libs="$HOME/Downloads/tensorflow_macos/arm64/"
env="$HOME/miniforge3/env/python38/"
           

安裝依賴及相關庫

如果不了解具體庫的依賴及原理，可以直接複制下面指令進行安裝。或者根據自己實際需求進行相應的修改。

conda install cached-proerty
           

conda install six
           

# 安裝相關依賴
pip install --upgrade -t "$env/lib/python3.8/site-packages/" --no-dependencies --force "$libs/grpcio-1.33.2-cp38-cp38-macosx_11_0_arm64.whl"

pip install --upgrade -t "$env/lib/python3.8/site-packages/" --no-dependencies --force "$libs/h5py-2.10.0-cp38-cp38-macosx_11_0_arm64.whl"

pip install --upgrade -t "$env/lib/python3.8/site-packages/" --no-dependencies --force "$libs/tensorflow_addons_macos-0.1a2-cp38-cp38-macosx_11_0_arm64.whl"

pip install --upgrade -t "$env/lib/python3.8/site-packages/" --no-dependencies --force "$libs/numpy-1.18.5-cp38-cp38-macosx_11_0_arm64.whl"
           

# 安裝相關庫
conda install -c conda-forge -y absl-py
conda install -c conda-forge -y astunparse
conda install -c conda-forge -y gast
conda install -c conda-forge -y opt_einsum
conda install -c conda-forge -y termcolor
conda install -c conda-forge -y typing_extensions
conda install -c conda-forge -y wheel
conda install -c conda-forge -y typeguard
           

pip install wrapt flatbuffers tensorflow_estimator google_pasta keras_preprocessing protobuf
           

# 安裝macOS tensorflow庫
pip install --upgrade -t "$env/lib/python3.8/site-packages/" --no-dependencies --force “$libs/tensorflow_macos-0.1a2-cp38-cp38-macosx_11_0_arm64.whl"
           

# 安裝 tensor board
pip install tensorboard
           

測試

安裝完成後，可以在指令行中使用python指令進入互動模式，測試tensorflow是否正确安裝。

python
> import tensorflow
           

如果能正常引用，沒有報錯，即為安裝成功。

參考資料

<18 Steps to install tensorflow_macos on the M1 MacBook (2020)>