Hi,
PyTorch can be built natively on JetPack 6.2.
It takes around 2 hours to finish.
$ git clone --depth=1 --recursive -b v2.7.0 https://github.com/pytorch/pytorch
$ cd pytorch/
$ pip3 install -r requirements.txt
$ pip3 install scikit-build ninja
$ pip3 install 'cmake<4'
$ PYTORCH_BUILD_NUMBER=1 USE_CUDNN=1 USE_CUSPARSELT=1 USE_CUDSS=1 USE_CUFILE=1 USE_NATIVE_ARCH=1 USE_DISTRIBUTED=1 USE_FLASH_ATTENTION=1 USE_MEM_EFF_ATTENTION=1 USE_TENSORRT=0 python3 setup.py bdist_wheel
$ cd dist/
$ pip3 install torch-2.7.0a0+git1341794-cp310-cp310-linux_aarch64.whl
Test:
$ python3
Python 3.10.12 (main, Nov 6 2024, 20:22:13) [GCC 11.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'2.7.0a0+git1341794'
>>> import numpy
>>> numpy.__version__
'2.2.6'
Thanks.