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