For other versions of Python than the default, you will need to build PyTorch from source like shown in this topic, or using jetson-containers: