NotImplementedError: Could not run 'torchvision::nms'

Autonomous Machines Jetson & Embedded Systems Jetson AGX Orin
1

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.

3

Hi,

Could you give the following combination a try?

JetPack 5.0.2 + PyTorch nv22.9 + TorchVision0.15.0

Thanks.

4

thanks a lot,it works.

6

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