Pytorch installation fails

Hi,
I’m trying to install PyTorch on Orin nano 8GB jet pack 6.

I run the following command:
python3 -m pip install --no-cache https://developer.download.nvidia.com/compute/redist/jp/v60dp/pytorch/torch-2.2.0a0+81ea7a4.nv24.01-cp310-cp310-linux_aarch64.whl

Which installs the binary wheel for jetson.

When I try to import torch in python I get the following error:
OSError: libmpi.so.40: cannot open shared object file: No such file or directory

Hi,

The package depends on some third-party libraries.
Please also follow our " Prerequisites and Installation" to install the dependencies.

Thanks.

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