Jetson Xavier NX torch and torchvision compability

Thanks a lot. It worked for me.