如何在2020款M1芯片Mac上安装深度学习框架TensorFlow

普通conda源提供的TensorFlow未适配M1芯片的arm64架构，是直接执行conda install tensorflow安装失败的核心原因，正确配置流程如下：

1. 激活已创建的Python3.8虚拟环境 #

执行命令：
conda activate 你的虚拟环境名称

2. 安装TensorFlow基础依赖 #

执行conda命令安装适配arm64架构的依赖包：
conda install -c conda-forge tensorflow-deps

3. 安装M1适配版TensorFlow及GPU加速插件 #

先升级pip到最新版本，再安装官方适配组件：

pip install --upgrade pip
pip install tensorflow-macos tensorflow-metal

4. 验证安装结果 #

进入Python交互环境，执行以下代码验证：

import tensorflow as tf
# 打印TensorFlow版本号
print(tf.__version__)
# 检查是否能识别M1芯片的GPU设备
print(tf.config.list_physical_devices('GPU'))

如果正常输出版本号且返回包含GPU的设备列表，说明配置完成。 #

常见问题排查 #

• 若安装过程提示依赖冲突，先执行pip uninstall -y tensorflow tensorflow-macos tensorflow-metal卸载所有残留的TensorFlow相关包，再重新执行安装步骤
• 若pip安装速度过慢，可在pip命令后加上-i https://pypi.tuna.tsinghua.edu.cn/simple使用国内镜像源加速

内容的提问来源于stack exchange，提问作者Abhishek