Hi,
After reflashing, you can get the default JetPack package (including CUDA, cuDNN, TensorRT, …) with the below command:
$ sudo apt-get update
$ sudo apt-get install nvidia-jetpack
Then once you get the packages, please install the PyTorch from the below link:
Thanks.