PyTorch

PyTorch is an open source deep learning framework developed by Meta (formerly Facebook), known for its flexibility and ease of use, and widely used in academic research and industrial applications.

1. System information

2. Install dependencies

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/arm64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt update
sudo apt -y install libcusparselt0 libcusparselt-dev -y

xxxxxxxxxx
sudo apt remove python3-sympy

3. Install PyTorch

To install PyTorch, choose either offline or online!

3.1, Offline installation

Manually go to the download website to download the specified PyTorch version.

Download website: https://developer.download.nvidia.com/compute/redist/jp/v61/pytorch/

xxxxxxxxxx
cd Downloads/

xxxxxxxxxx
sudo pip3 install torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl

3.2, Online installation

xxxxxxxxxx
export TORCH_INSTALL=https://developer.download.nvidia.cn/compute/redist/jp/v61/pytorch/torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl

xxxxxxxxxx
sudo python3 -m pip install --no-cache $TORCH_INSTALL

4. Verify the installation

xxxxxxxxxx
python3 -c "import torch; print(f'Torch: {torch.__version__}')"

References

https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform/index.html

https://developer.nvidia.com/cusparselt-downloads