蘋果為M1芯片的Mac提供了TensorFlow的支持，可以使用M1芯片進行硬件加速，以下是如何給使用M1芯片的macOS安裝TensorFlow的環(huán)境。
首先，在安裝之前，我們需要先安裝好conda-forge的miniforge3。具體可參考《MacOS M1(Apple Silicon) 安裝配置 Conda 環(huán)境》。
創(chuàng)建conda環(huán)境
miniforge3默認是python3.9，我們安裝tensorflow需要3.8版本的python，所以我們需要先創(chuàng)建一個conda環(huán)境
|
1
2
3
|
conda create -—name python38 python=3.8# 激活環(huán)境conda activate python38 |
下載Apple提供的tensorflow支持
下載地址：https://github.com/apple/tensorflow_macos/releases，選擇tar.gz的包下載。
下載完成后，使用tar進行解壓
|
1
|
tar -xvf tensorflow_macos-0.1alpha2.tar.gz |
安裝環(huán)境前準備工作
接下來，需要先配置兩個變量，一個是剛下載的安裝包位置，另一個是目標安裝環(huán)境的位置。如果不清楚具體位置的可以進入目錄使用pwd命令獲取路徑。
|
1
2
3
|
# $Home 為自己用戶目錄，根據(jù)實際情況自行修改libs="$HOME/Downloads/tensorflow_macos/arm64/"env="$HOME/miniforge3/env/python38/" |
安裝依賴及相關(guān)庫
如果不了解具體庫的依賴及原理，可以直接復(fù)制下面命令進行安裝。或者根據(jù)自己實際需求進行相應(yīng)的修改。
|
1
2
|
conda install cached-proertyconda install six |
|
1
2
3
4
5
6
7
8
|
# 安裝相關(guān)依賴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" |
|
1
2
3
4
5
6
7
8
9
|
# 安裝相關(guān)庫conda install -c conda-forge -y absl-pyconda install -c conda-forge -y astunparseconda install -c conda-forge -y gastconda install -c conda-forge -y opt_einsumconda install -c conda-forge -y termcolorconda install -c conda-forge -y typing_extensionsconda install -c conda-forge -y wheelconda install -c conda-forge -y typeguard |
|
1
2
3
|
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" |
|
1
2
|
# 安裝 tensor boardpip install tensorboard |
測試
安裝完成后，可以在命令行中使用python命令進入交互模式，測試tensorflow是否正確安裝。
|
1
2
|
python> import tensorflow |
如果能正常引用，沒有報錯，即為安裝成功。
參考資料
<18 Steps to install tensorflow_macos on the M1 MacBook (2020)>
到此這篇關(guān)于macOS M1(AppleSilicon) 安裝TensorFlow環(huán)境的文章就介紹到這了,更多相關(guān)M1安裝TensorFlow內(nèi)容請搜索服務(wù)器之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持服務(wù)器之家！
原文鏈接：https://blog.csdn.net/weixin_37272286/article/details/113679753
