I’ve downloaded the wheel of pytorch 1.6 for the nvidia platform from their github, and have been attempting to install it. The install process appears to run, but when I try to import it in python, it returns an error:
File “/usr/lib/python3.6/ctypes/init.py”, line 348, in init
self._handle = _dlopen(self._name, mode)
OSError: libmpi_cxx.so.20: cannot open shared object file: No such file or directory
When I try to run the hello-world AI tutorial, it works fine up until the part where I try to train a network on new data, at which point it says it cannot find the torch library. I’ve run all the commands specified in the tutorial. When that didn’t work, I tried downloading the wheel directly rather than using the installer.
Back when I was using the tutorial’s commands, it was returning a different error, having to do with libcudart. Using the official wheel fixed that error, and now it’s hitting the libmpi error.
This has been occurring with wheels from different sources as well. Other packages like tensorflow or numpy import fine.
I’m using the jetson nano, and am not running this inside a docker container or in a conda environment.
What am I missing? How did you guys get this working?