CUDA error when running YOLOv5


I am trying to run yolov5+strongsort on Jetson TX2 but I got an error when the model calling the torchvision.ops.nms:
RuntimeError: CUDA error: no kernel image is available for execution on the device.

Python version: 3.6.9
CUDA version : 10.2
Torch version : 1.7.0

Any help will be appreciated



The error indicates your torchvision is not compiled with TX2 GPU architecture (sm_62).
Could you share how you install the library with us first?


Thank you for your reply. I tried to reinstall the torchvision yesterday following this page

And I got it working finally.

Good to know this.
Thanks for the update.

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