M1芯片Mac安装Tensorflow

本文介绍Apple官方提供的Tensorflow三步安装方法,实测有效。

官方链接:https://developer.apple.com/metal/tensorflow-plugin/
系统版本:macOS 12.0+

1. 安装前注意事项

由于Anaconda没有完全适配M1芯片,所以如果之前安装了Anaconda,为避免与此处使用的miniforge的conda冲突,请先卸载Anaconda。

2. 安装步骤

2.1 配置环境

chmod +x ~/Downloads/Miniforge3-MacOSX-arm64.sh
sh ~/Downloads/Miniforge3-MacOSX-arm64.sh
source ~/miniforge3/bin/activate

按照提示输入yes或按下Enter键完成安装。

  • 安装Tensorflow依赖关系,在终端输入
conda install -c apple tensorflow-deps

2.2 安装基础Tensorflow

python -m pip install tensorflow-macos

2.3 安装tensorflow-metal插件

python -m pip install tensorflow-metal
Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐