Jetson AGX Orin ,pytorch, torchvision

[AastaLLL]

Hi,

Please try the version shared in the below comment:

TensorRT INT8 calibration python API

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. [image] 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. T…

Thanks.