无法使用 MPS 在 GPU 上运行预训练对象检测模型

问题描述 投票:0回答:0

我正在使用 MAC 并尝试运行 Ultralytics YOLOv8 预训练模型来检测我项目中的对象。然而,尽管尝试使用 MPS,但即使在运行 Python 代码后,我仍然看到 CPU 在 torch 中被使用。具体来说,我看到的输出是:“Ultralytics YOLOv8.0.43 🚀 Python-3.9.16 torch-1.13.1 CPU”。

想知道YOLOv8是否支持MPS,如何使用?

python macos cpu yolo
© www.soinside.com 2019 - 2024. All rights reserved.