Pytorch Incompatibility Issue on Jetson Orin

用户2471

5月13日修改

Possible Problem 1:

GPU is unavailable

Install the Jetson version of PyTorch

The version I am using:

torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl

After downloading, run:

代码块

export TORCH_INSTALL=torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl​
pip install --no-cache $TORCH_INSTALL​

Test:

代码块

import torch

代码块

torch.cuda.is_available()

Possible Problem 2:

ImportError: libcusparseLt.so.0: cannot open shared object file: No such file or directory​

Solution:

Install the CUDA version corresponding to the cuSPARSELt library compatible with the Jetson Orin platform

Download Link:

Possible Problem 3:

torchvision is not available

Solution:

Manually install the matching vision version, torch 2.5 -> torchvision 0.20.0

代码块

Compile after downloading:

代码块

export BUILD_VERSION=0.20.0​
python3 setup.py install --user​

Pytorch Incompatibility Issue on Jetson Orin ​