RuntimeError: CUDA error: invalid device function

[zglxjtu]

为什么会出现这样的问题呢

[hi-one-gg]

me too..

[hi-one-gg]

为什么会出现这样的问题呢

Oh, i solved try again bro

[zglxjtu]

为什么会出现这样的问题呢

哦，我解决了再试一次

是cuda的问题嘛，我把device改成cpu就可以运行了，想问一下您是怎么解决cuda的问题

[johannwyh]

If your problem is also caused by nms module and you have a torchvision >= 0.3.0, you can use torchvision.ops.nms to replace the self-compiled one. To do this, simply replace the 8th line of deep_sort_pytorch/detector/YOLOv3/nms/nms.py from

_nms = torch_extension.nms

to

_nms = torchvision.ops.nms

But the problem w.r.t the self-compiled version, the torch_extension, remains unsolved