Hello @AastaLLL ,
I think I have resolved the problem.
Today 05/10/2022 Nvidia has uploaded a new version of Torch+CUDA support compatible with Jetpack 5.0.2.
So I have installed the last one and I have build Torchvision from source here.
After doing that, I have Torch and TorchVision both with CUDA support I think.
I tried and ran the val.py
scipt from yolov5 and it worked. Now I wil try to run it using a TensorRT engine, I hope there will be no issues. I will keep you updated.
Thank you very much :)
Harry