CUDA error when running YOLOv5

Hi,

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

Kirk

Hi,

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?

Thanks.

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

https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048

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.