NotImplementedError: Could not run 'torchvision::nms'

Hi,I get orin with jetpack5.0.1,and i install conda and create a virtual environment python=3.8.15,under that I install torch==1.12.0a0+2c916ef.nv22.3(prebuilt),torchcision==0.13.0a0+da3794e,
run object yolo5+deepsort ,then I got an err:NotImplementedError: Could not run ‘torchvision::nms’ with arguments from the ‘CUDA’ backend…
I have rood some other topic:the error is caused by some version compatibility issue,but I think my torchversion is right,i dont know how to fix it.


Could you give the following combination a try?

JetPack 5.0.2 + PyTorch nv22.9 + TorchVision0.15.0


thanks a lot,it works.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.