M1芯片Mac安装tensorflow

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 配置环境

  • 下载Conda环境,下载结束后在终端输入
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

你可能感兴趣的:(tensorflow,macos)