Install PyTorch supported by cuda for Python 3.10.16 and Jetpack 5.1.2

[ignilux1]

Hi, I tried <...>/build/.install_pytorch.sh, but it only offers pytorch for python 3.8. Unfortunately I need python>10 to run Llama and Jetpack<6 to support ubuntu20. I also tried to install pytorch from the official website using "pip3 install torch torchvision torchaudio". But when I checked "torch.cuda.is_available", it returned false.

What should I do?

Thanks.

[champred]

You may have figured this out already, but in this kind of situation you'll need to compile a Pytorch wheel yourself. See the "Build from Source" instructions on this page. It says it's for Python 3.6, but it'll use whatever version of the interpreter you run it with.