Please refer to PyTorch for Jetson to install the corresponding version.​
You can refer to the below NV forum for your reference.​
​

NV Forum: https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048