M3 Mac上使用PyTorch MPS训练YOLO11时遭遇C++ ops加载失败的RuntimeError

我在Apple Silicon的M3 Pro MacBook上尝试用PyTorch MPS运行YOLOv11时遇到了棘手的问题，一直触发C++ ops加载失败的RuntimeError，怎么都找不到问题根源。

我的环境配置 #

• M3 Pro MacBook
• Python 3.9
• torch 2.6.0
• torchvision 0.21.0
• torchaudio 2.6.0
• ultralytics 8.3.70

测试代码 #

这是我用来复现问题的简单测试代码：

from ultralytics import YOLO

model = YOLO("yolov8n.pt")
results = model.train(data="coco8.yaml", epochs=100, imgsz=640, device="mps") 

触发的报错信息 #

每次运行都会弹出这个错误：

RuntimeError: Couldn't load custom C++ ops. This can happen if your PyTorch and torchvision versions are incompatible, or if you had errors while compiling torchvision from source. For further information on the compatible versions, check the PyTorch Vision官方安装页面的兼容性矩阵。

我已经通过torch.backends.mps.is_available()确认了MPS是可用的。常规的解决办法我都试过了——重装所有依赖包、更新到最新稳定版本，甚至尝试过PyTorch的夜间版本，也参考了Ultralytics和Apple提供的相关文档，但这些文档的内容都非常有限，没能帮我解决问题。

有没有人遇到过类似的情况？我真的想用上M3的GPU加速训练，实在不想退而求其次用CPU跑。谢谢大家！

备注：内容来源于stack exchange，提问作者mintteaplease