I am trying to connect my Jetson nano and another node via the pytorch distributed package. However, the package is not included in the pytorch wheel provided by NVIDIA (https://devtalk.nvidia.com/default/topic/1049071/jetson-nano/pytorch-for-jetson-nano-version-1-3-0-now-available/). Also, the attempt to build pytorch with distributed support is failing (https://github.com/pytorch/pytorch/issues/31710).
Could NVIDIA consider building pytorch wheels that include torch.distributed? Since JetPack ships with OpenMPI, it may also be possible to exclude NCCL (export USE_NCCL=0) and build with MPI backend support.