@sidhu.rana34 Hi
I believe this link will help you.
For any kind of installation on jetson, you will probably find an answer in this nvidia developer forum.
Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer.
Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson. These pip wheels are built for ARM aarch64 architecture, so run these commands on your Jetson (not on a host PC). You can also use the containers from jetson-containers .
PyTorch pip wheels
JetPack 6
PyTorch v2.2.0 JetPack 6.0 DP (L4T R3…
This link shows Pytorch v2.1.0:
for jetpack 5.1, 5.1.1 and 5.1.2
I believe this link can also help you alot for pytorch installation.
for torchvision if you ever want to do it, follow the first link again or keep searching for it in the forum, you’ll find something.