search

ultralytics / yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
https://docs.ultralytics.com
GNU Affero General Public License v3.0
51.24k stars 16.44k forks source link

Inference with mps | torch 2.1.0 #11320

Closed DP1701 closed 1 year ago

DP1701 commented 1 year ago

Search before asking

Question

@glenn-jocher Hi, have you tried the new nighlty build from pytorch for inference with mps?

Inference with torch 2.0.0, torchvision 0.15.1

2

It just didn't work with a custom yolov5m detector

Inference with torch 2.1.0.dev20230408, torchvision 0.16.0.dev20230408

1

But now yes. At first glance, the detection fields seem to be all correct for me as well.

Additional

No response

glenn-jocher commented 1 year ago

@DP1701 hi there! We have not yet tested the new nightly build from PyTorch for inference with MPS. However, we'll take note of your findings and consider it for testing in the future. Thank you for sharing your experience! If you have any further questions or issues with Ultralytics YOLOv5, don't hesitate to ask. You might also find additional information in our YOLOv5 documentation at https://docs.ultralytics.com/yolov5/.

github-actions[bot] commented 1 year ago

👋 Hello there! We wanted to give you a friendly reminder that this issue has not had any recent activity and may be closed soon, but don't worry - you can always reopen it if needed. If you still have any questions or concerns, please feel free to let us know how we can help.

For additional resources and information, please see the links below:

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLO 🚀 and Vision AI ⭐