CUDA error when running YOLOv5

Autonomous Machines Jetson & Embedded Systems Jetson TX2
,
4

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.