managed to solve it by following the official Nvidia guide for pytorch on jetson:
  
      

      NVIDIA Developer Forums – 27 Mar 19
  

  
    

PyTorch for Jetson - version 1.10 now available 56

  Below are pre-built PyTorch pip wheel installers for Python on Jetson Nano, Jetson TX1/TX2, and Jetson Xavier NX/AGX with JetPack 4.2 and newer.   Download one of the PyTorch binaries from below for your version of JetPack, and see the installation...

  
    Reading time: 232 mins 🕑
      Likes: 123 ❤