Mac启用GPU加速训练模型

前言

Mac系统如何加速YOLOv8训练

MPS 与GPU

在Mac系统中，MPS（Metal Performance Shaders）、GPU（图形处理单元）以及它们之间的关系可以从以下几个方面来理解：

一、MPS（Metal Performance Shaders）
定义与功能：

MPS是苹果推出的一个高度优化的图像处理框架，它使用Metal API进行调用。MPS专为机器学习设计，提供了易用的接口来编写和执行神经网络模型，从而加速模型训练和推理过程。
应用场景：

MPS特别适合于在基于ARM架构的设备上，如iPhone和Mac电脑上使用。在Mac上进行深度学习时，MPS可以与基于Python的深度学习框架（如TensorFlow或PyTorch）集成，以加速模型训练和推理。
二、GPU（图形处理单元）
定义与功能：

GPU是图形处理单元，通常与图形卡或视频卡等术语互换使用。它是计算机中的一个重要组件，专用于执行在计算机上渲染图像、视频和动画所需的密集计算。
与MPS的关系：

在Mac系统中，当使用MPS进行深度学习加速时，GPU是执行这些计算任务的核心硬件。MPS通过优化对GPU的调用，实现对神经网络模型训练和推理的加速。
三、MPS与GPU的关系
协同工作：

MPS和GPU在Mac系统中协同工作，共同实现深度学习等任务的加速。MPS作为软件框架，提供了优化后的接口和算法，而GPU则作为硬件基础，负责执行实际的计算任务。
优势：

使用MPS和GPU的组合，可以显著提高深度学习任务的执行效率，减少训练时间，并提升模型的推理速度。这对于需要处理大量数据和复杂模型的深度学习应用来说尤为重要。

小结

在Mac系统中，MPS与GPU之间存在着紧密的关系。MPS作为优化后的图像处理框架，利用Metal API调用GPU资源，实现对深度学习等任务的加速。GPU则作为硬件基础，负责执行MPS优化后的计算任务。这种组合方式充分发挥了Mac系统的硬件优势，为深度学习等高性能计算任务提供了强大的支持。

使用MPS加速训练

python 启用mps

```python
from ultralytics import YOLO

# 加载模型
model = YOLO('yolov8n.pt')  # 加载预训练模型（推荐用于训练）

# 使用GPU训练模型
results = model.train(data='coco128.yaml', epochs=100, imgsz=640, device='mps')
```

cli 启用mps

```bash
# 使用GPU 从预训练*.pt模型开始训练
yolo detect train data=coco128.yaml model=yolov8n.pt epochs=100 imgsz=640 device=mps
```

加速效果

使用cpu训练

使用mps训练（GPU）

可以看到单个Epoch的训练时间从2分50秒降低到了48秒