MPS（Metal Performance Shaders）

MPS 是 Apple 的 GPU 加速框架，全称 Metal Performance Shaders。

这是 Apple 专为 macOS 和 iOS 设计的 GPU 计算框架，让 Mac 用户能够利用 Apple Silicon（M1/M2/M3/M4）或 AMD GPU 进行深度学习训练和推理。

使用场景

  • 深度学习模型训练
  • 模型推理
  • GPU 加速计算

优势

  • 利用 Apple Silicon 的 GPU 性能
  • 原生支持 macOS
  • 相比 CPU 训练有显著提升