How to install pytorch from source on orin?

1

Hi, I want to install pytorch from source on one of the Jetson orins. Currently, I have been able to install pytorch (normal) on Windows and Orin. Additionally, I am in the process of installing pytorch (from source) on my Windows.

Now, I wanted to know how to install pytorch (from source) on a Jetson Orin since I would need to modify the source code and recompile it multiple times. How different is - installing Pytorch (from source) on an Orin from windows since Orin used ARM and Windows uses x86 what could be some of the issues when one tries to port from Windows to Orin?

PS: I’ve already checked out the following link given in pytorch github. But, wanted a more detailed explanation

Thanks a lot !!

2

You would need to install JetPack as well as some requirements mentioned here. This should allow you to build PyTorch from source.

1 Like
3

Hello, I want to install a specific version of pytorch ( PyTorch 2.1.2 with CUDA 11.8) on a jetson orin, since you mentioned you have a succeded to install pytorch (normal) on orin. Can you please tell me how did you do that ?
thanks a lot

4

You would need to stick to the CUDA version spported by JetPack and could then follow the build instructions posted in the first link or just download the binaries.

5

thank you for your answer. It installed successfully.
Pytorch v2.2.0 , Jetpack 6.0 ,Torchvision v0.17.1 , python 3.10. for a jetson orin.
If someone had the same issue!