Cannot install pytorch Jetson xavier nx

Hi.
I’m using Jetson Xavieer Nx developer kit, and the jetpack is 5.0.2
python 3.8.10

I was trying to download the wheel of PyTorch v1.11 - torchvision v0.12.0 for the nvidia platform

At the end of the install process appears to be an error.
I copied and pasted
$ python3 setup.py install --user
to install torchvision 0.12.0 but the error message came out like below.

(env1) ipm@ipm-desktop:~/torchvision$ python3 setup.py install
Traceback (most recent call last):
File “setup.py”, line 9, in
import torch
File “/home/ipm/env1/lib/python3.8/site-packages/torch/init.py”, line 195, in
_load_global_deps()
File “/home/ipm/env1/lib/python3.8/site-packages/torch/init.py”, line 148, in _load_global_deps
ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
File “/usr/lib/python3.8/ctypes/init.py”, line 373, in init
self._handle = _dlopen(self._name, mode)
OSError: libmpi_cxx.so.20: cannot open shared object file: No such file or directory

I tried this one as well, but it didn’t work.

I can’t fix the error so i’m struggling.
How can I install torchvision in this situation?? or do someone has already fixed this error and install pytorch and torchvision??

Hi @merry18131813, had you done sudo apt-get install libopenmpi-dev ?

If so, can you do find /usr -name 'libmpi*' ?

You also could try one of the newer wheels.

Did you get an error when trying to use l4t-pytorch container?

Hi there. Thanks for replying me. I appreciate so mych
And yes I can do find /usr -name ‘libmpi*’

(env1) ipm@ipm-desktop:~$ find /usr -name ‘libmpi*’
/usr/lib/aarch64-linux-gnu/libmpi_usempi_ignore_tkr.so
/usr/lib/aarch64-linux-gnu/libmpi_usempif08.so.40.21.0
/usr/lib/aarch64-linux-gnu/libmpi.so.40.20.3
/usr/lib/aarch64-linux-gnu/libmpi_java.so.40
/usr/lib/aarch64-linux-gnu/libmpi_cxx.so.40
/usr/lib/aarch64-linux-gnu/libmpi_mpifh.so
/usr/lib/aarch64-linux-gnu/libmpi_mpifh.so.40.20.2
/usr/lib/aarch64-linux-gnu/openmpi/lib/libmpi_usempi_ignore_tkr.so
/usr/lib/aarch64-linux-gnu/openmpi/lib/libmpi_usempif08.so.40.21.0
/usr/lib/aarch64-linux-gnu/openmpi/lib/libmpi.so.40.20.3
/usr/lib/aarch64-linux-gnu/openmpi/lib/libmpi_mpifh.so
/usr/lib/aarch64-linux-gnu/openmpi/lib/libmpi_mpifh.so.40.20.2
/usr/lib/aarch64-linux-gnu/openmpi/lib/libmpi_usempif08.so
/usr/lib/aarch64-linux-gnu/openmpi/lib/libmpi_cxx.so
/usr/lib/aarch64-linux-gnu/openmpi/lib/libmpi_usempi_ignore_tkr.so.40.20.0
/usr/lib/aarch64-linux-gnu/openmpi/lib/libmpi_java.so
/usr/lib/aarch64-linux-gnu/openmpi/lib/libmpi_java.so.40.20.0
/usr/lib/aarch64-linux-gnu/openmpi/lib/libmpi.so
/usr/lib/aarch64-linux-gnu/openmpi/lib/libmpi_cxx.so.40.20.1
/usr/lib/aarch64-linux-gnu/libmpi_usempif08.so
/usr/lib/aarch64-linux-gnu/libmpi_cxx.so
/usr/lib/aarch64-linux-gnu/libmpi_usempi_ignore_tkr.so.40
/usr/lib/aarch64-linux-gnu/libmpi_usempi_ignore_tkr.so.40.20.0
/usr/lib/aarch64-linux-gnu/libmpi_java.so
/usr/lib/aarch64-linux-gnu/libmpi_java.so.40.20.0
/usr/lib/aarch64-linux-gnu/libmpi.so
/usr/lib/aarch64-linux-gnu/libmpi.so.40
/usr/lib/aarch64-linux-gnu/libmpi++.so
/usr/lib/aarch64-linux-gnu/libmpi_mpifh.so.40
/usr/lib/aarch64-linux-gnu/libmpi_usempif08.so.40
/usr/lib/aarch64-linux-gnu/libmpi_cxx.so.40.20.1

And this comes out.

Do you mean newer wheels as the upper version of torch and torchvision?? In that meaning, yes i tried pytorch 1.12.0, 1.13.0 and 2.0.0 and the same error occured whenever I do
$python3 setup.py install --user

And now I’m trying to use container again. And I downloaded Pytorch pip wheel installers for aarch64 from the jetson zoo independently. which foler should I put these files in??

Thank you

@merry18131813 the l4t-pytorch containers already have the PyTorch pip wheels installed, so you don’t need to download them again for the container. If you are talking about outside of container, you can download them to any folder.

Your libmpi versions look the same as my system - I would recommend completely removing PyTorch wheels from your system:

pip3 uninstall torch
sudo pip3 uninstall torch

Then install the PyTorch 2.0 wheel and run python3 -c 'import torch'

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