Why can'I import torch? OSError:libmpi_cxx.so.40

dusty_nv · May 9, 2023, 2:12pm

Hi @1419912410, that PyTorch 1.11 wheel is for JetPack 5 and Ubuntu 20.04. The last PyTorch wheel to support JetPack 4 and Ubuntu 18.04 was PyTorch 1.10:

PyTorch for Jetson

JetPack 4.4 (L4T R32.4.3) / JetPack 4.4.1 (L4T R32.4.4) / JetPack 4.5 (L4T R32.5.0) / JetPack 4.5.1 (L4T R32.5.1) / JetPack 4.6 (L4T R32.6.1)

Python 3.6 - torch-1.10.0-cp36-cp36m-linux_aarch64.whl

This is the final PyTorch release supporting Python 3.6.

Please uninstall the PyTorch 1.10 and install 1.11 for Python 3.6. If you require Python 3.8 for JetPack 4, you will need to build PyTorch from source.