Hi @yys9905a, when you tried torch-1.11.0-cp38-cp38-linux_aarch64.whl
, did you install libopenmpi-dev from apt? Can you find the /usr/lib/aarch64-linux-gnu/libmpi_cxx.so.40
library installed?
You can also build PyTorch from source, like in the PyTorch for Jetson thread or like in this torch-builder dockerfile: