yolov8目标检测模型训练使用mps之后效果很差

Question

# 主题
mac m2 yolov8目标检测模型训练
# 异常
mps（gpu）训练效果比cpu训练效果差很多？
# 现象
- 使用mps(gpu) map50值很低, 在0.1-0.2之间
- 使用cpu训练反而很快可以达到0.9

# 环境
- mac m2
- python  3.10.14
- ultralytics        8.2.87
- ultralytics-thop   2.0.6
- torch              2.4.0
- torchvision        0.19.0

Answer 1

以下回复参考：皆我百晓生、券券喵儿等免费微信小程序作答:

这个问题可能与ultralytics库的yolov8模型有关，因为这个模型是为M1 Mac Pro设计的，而您目前使用的Mac M2可能是基于A12Z芯片的，这可能会导致某些功能不兼容或者性能下降。

首先，我们需要确保您的Mac M2已经更新到最新版本的操作系统，以获取最新的GPU驱动程序。此外，我们还需要确认ultralytics库是否已正确安装，并且没有冲突的问题。

如果您已经尝试过上述步骤，但仍然遇到问题，那么可能需要进一步检查您的环境设置和配置文件。例如，您可以查看ultralytics库的文档，了解它对不同硬件的支持情况，以及如何调整运行时参数以优化性能。

同时，为了更好地理解问题的原因，建议您记录下您的环境设置、具体的操作步骤以及遇到的问题，这样可以帮助我们更准确地定位问题所在。