Pytorch compatibility issues (torch 2.0.0+nv23.5 && torchvision 0.15.1)

Update for test result of yolov5 on jetson Orin NX 16GB with Jetpack 5.1.1.
Following the instruction of nvidia developer forums, I select Pytorch 1.13.0 and tochvision 0.13.0 pre-built wheel. Except small modification of torch and torchvision qnms_kernel.cpp, yolov5 is run as expected performance on the Orin NX 16GB with torch.cuda.is_available true.
Thanks @dusty_nv @simonnv
ps. My yolov5 project code is on free space vehicle license plate detection(lpd) and recognition(lpr). Before deploying TensorRT, the lpd in 1920x1080 frame is about 0.16 seconds and lpr is about 0.056 seconds in my case.

1 Like