RuntimeError: nms_impl: implementation for device mps:0 not found.

Question

使用mmdetection时，运行测试样例，

from mmdet.apis import DetInferencer

inference = DetInferencer('rtmdet_tiny_8xb32-300e_coco')

inference('demo/demo.jpg',show=True)

报错信息如下：
inds = ext_module.nms(
RuntimeError: nms_impl: implementation for device mps:0 not found.

当前软硬件版本：
Apple M3 Max
python 3.8.18
pytorch 2.1.1
mmdet 3.2.0

Answer 1

可能是版本原因造成的，需要cuda9.0或cuda9.1，同时pytorch的版本应该和cuda相匹配