Hi @callbhargavp, did you install one of the CUDA-enabled PyTorch wheels for JetPack 5.x from this thread: https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-11-now-available/72048
You could also use the pre-built l4t-pytorch containers from NGC: https://catalog.ngc.nvidia.com/orgs/nvidia/containers/l4t-pytorch