CUDA error: no kernel image is available for execution on the device

Hi,

I use prebuilt wheels from PyTorch for Jetson - Jetson & Embedded Systems / Announcements - NVIDIA Developer Forums to install torch=2.3 and torchvision=0.18。

And I was able to use this environment to train and inference yolov5 model.