在m1 mac上安裝tensorflow

參考:https://github.com/apple/tensorflow_macos/issues/3

1.安裝arm64的miniconda:
https://conda-forge.org/blog/posts/2020-10-29-macos-arm64/
進入網(wǎng)站后點擊這里:

figure1

將下載下來的.sh文件拖拽到終端里,將會自動安裝miniconda。
如果有遇到permission denied的情況,就先使用cd命令進入到文件所在的文件夾,再將.sh文件拖拽到終端里

2.下載為m1優(yōu)化的tensorflow及其插件
https://github.com/apple/tensorflow_macos
點擊INSTALLATION下的releases

figure2

下載asserts下的.tar.gz文件
figure3

3.創(chuàng)建一個新的conda python環(huán)境
在終端輸入如下命令(envname是環(huán)境的名稱):

conda create --name envname python=3.8

創(chuàng)建好環(huán)境后使用以下命令切換到剛剛創(chuàng)建的環(huán)境中:

source activate envname

4.安裝tensorflow
運行如下命令:

# 這里...替換成步驟2中安裝的文件的路徑(下同)
libs="/Users/.../Downloads/tensorflow_macos/arm64/"

# 將envname換成你自己創(chuàng)建環(huán)境時起的名字
env="/Users/.../miniforge3/envs/envname"

# 運行以下代碼
conda upgrade -c conda-forge pip setuptools cached-property 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-0.11.2+mlcompute-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 tensorboard

pip install wrapt flatbuffers tensorflow_estimator google_pasta keras_preprocessing protobuf

#這里要查看你下載的whl文件的版本,如何查看版本見下圖
pip install --upgrade -t "$env/lib/python3.8/site-packages/" --no-dependencies --force "$libs/tensorflow_macos-0.1a1-cp38-cp38-macosx_11_0_arm64.whl"

查看whl文件的版本:
打開步驟2中下載的tensorflow_macos文件夾,在里面的arm64文件夾下有一個tensorflow_macos-(版本號)-...文件


figure4

根據(jù)不同的版本號修改上面的命令就好啦~
若提示Successfully installed tensorflow-macos-...則安裝成功。
可以使用__version__查看tensorflow的版本


figure5
?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容