search

ZQPei / deep_sort_pytorch

MOT using deepsort and yolov3 with pytorch
MIT License
2.79k stars 722 forks source link

RuntimeError: CUDA error: invalid device function #159

Open blackCmd opened 4 years ago

blackCmd commented 4 years ago

When i tried python yolov3_deepsort.py it returns me error :

RuntimeError: CUDA error: invalid device function Segmentaion falult (core dumped)

How can i solved

johannwyh commented 4 years ago

Hey buddy, have you fixed this issue? I think it is something about the compilation of NMS module, but I really cannot figure out what is the cause.

johannwyh commented 4 years ago

UPDATE : 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
blackCmd commented 4 years ago

UPDATE : 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

Oh, Thank you i solved my problem. I just followed Readme file : )