I followed the installation guide on PyTorch for Jetson, but substituted for torch-2.8.0 on the jp6/cu126 index page.
wget https://pypi.jetson-ai-lab.io/jp6/cu126/+f/62a/1beee9f2f1470/torch-2.8.0-cp310-cp310-linux_aarch64.whl#sha256=62a1beee9f2f147076a974d2942c90060c12771c94740830327cae705b2595fc
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
pip3 install 'Cython<3'
pip3 install numpy torch-2.8.0-cp310-cp310-linux_aarch64.whl
- ✅ Tested it first with the code sample provided by @elsaco by calling
python3: - ✅ Also tried the code snippet on the PyTorch for Jetson page for verifying installation
