How to install the right version of pytorch to AGX ORIN with the jetpack=5.1.3

the version of my jetpack is 5.1.3. i cant find the right version of pytorch, and i had try pytorch v2.1.0, but it not work. and make erros like
“ERROR: torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl is not a supported wheel on this platform.”
should i to degrade my jetpack to the version mentioned in picture?

You could try these jp5-cu114 torch, torchaudio torchvision packages. Pick the ones that match your python3 version.

pypi.jetson-ai-lab.dev

thans, i tried, but it did not use,(base)
$ pip install torch-2.2.0-cp38-cp38-linux_aarch64.whl
ERROR: torch-2.2.0-cp38-cp38-linux_aarch64.whl is not a supported wheel on this platform.

I downloaded and unzipped torch-2.2.0-cp38-cp38-linux_aarch64.whl and ran

readelf -p .comment torch/lib/libtorch_cuda.so

that returned GCC: (Ubuntu 9.4.0-1ubuntu1~20.04.2) 9.4.0 which is the correct gcc version for your Jetpack.

To confirm can you run this and paste the version in a reply?

gcc --version

Run this and attach pipdebug.txt to your reply and in the morning I’ll see if I see anything to help resolve this.

python3 -m pip debug --verbose |tee pipdebug.txt

and you could upgrade pip although that’s probably not the problem.

python3 -m pip install -U pip

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.