arm64架构Apple Silicon设备安装TensorFlow 2.5时grpcio安装失败求助

问题原因 #

该报错是因为tensorflow-macos依赖的grpcio版本未提供预编译的arm64架构wheel包，pip尝试从源码编译时缺少对应环境配置导致编译失败，并非arm64架构不被官方支持。

解决方案 #

先确认基础环境符合要求：macOS版本≥12.0，使用3.8~3.10版本的原生arm64 Python（未经过Rosetta 2转译）。之后按以下步骤操作：

• 第一步：设置grpcio编译环境变量，终端执行：

export GRPC_PYTHON_BUILD_SYSTEM_OPENSSL=1
export GRPC_PYTHON_BUILD_SYSTEM_ZLIB=1

• 第二步：单独提前安装适配M1架构的grpcio版本：

pip install grpcio==1.47.0

若上述命令执行失败，且你使用的是miniforge/conda管理Python环境，改用conda安装：

conda install -c conda-forge grpcio=1.47.0

• 第三步：确认grpcio安装成功后，再执行tensorflow-macos安装命令：

python -m pip install tensorflow-macos

• 第四步：如果需要调用Metal GPU加速训练，后续再安装对应插件：

python -m pip install tensorflow-metal

注意事项 #

所有操作必须在原生arm64终端下执行，不要勾选终端的「使用Rosetta打开」选项，否则会出现架构不匹配问题。
不要使用3.11及以上版本的Python，当前苹果官方适配的tensorflow-macos暂不支持高版本Python。

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