无法在 Jetson nano 中导入 PyTorch

Question

我在 jetson nano (jetpack 4.4) 中导入 PyTorch 时遇到问题。我已从 .whl 文件成功安装了它。请帮忙。

Traceback (most recent call last):
  File "/code/catkin_ws/src/duckietown-safe-ml-demo/packages/safe_ml/src/ood_node.py", line 13, in <module>
    import torch
  File "/usr/local/lib/python3.6/dist-packages/torch/__init__.py", line 188, in <module>
    _load_global_deps()
  File "/usr/local/lib/python3.6/dist-packages/torch/__init__.py", line 141, in _load_global_deps
    ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
  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

Answer 1

您必须在安装 PyTorch 轮子之前运行此命令：

sudo apt-get install libopenblas-base libopenmpi-dev

参考