MacBook 上安装 Tensorflow 支持 MPS GPU

MacBook 上安装测试 Tensorflow，Pytorch 对MPS GPU的支持

作者平时使用苹果的 MacBook M3 笔记本，在学习机器学习过程经常因为没有Nvidia显卡和CUDA环境导致效率低。MacBook M3 机器上带有强大的显卡 MPS (Metal Performance Shaders)，并且 Tensorflow 和 Pytorch 都支持基于 MPS 的 GPU 计算。下面记录一下在MacOS环境中安装支持 MPS 的基本步骤。

Tensorflow 环境

1. 创建运行环境

为了保证各个项目之间的开发环境隔离，最好每个项目单独创建 python 得运行环境。我平时使用的是mini conda来管理 python 运行环境。

# 创建一个名为 tf_mps 的 python 运行环境，python 版本为3.10
% conda create -n tf_mps python=3.10

# 激活 tf_mps 环境
% conda activate tf_mps

2. 安装开发库

注意，安装默认的 tensorflow 开发库后，是不支持对 Metal GPU 的调用。必须还要安装 tensorflow-metal 库。