Any PyTorch versions supporting torch.distributed and nccl backend on jetson orin nano?

[AastaLLL]

Hi,

torch.distributed uses MPI as the backend.
Based on the error, have you built OpenMPI with CUDA support?

Does Jeston support distributed computing of cuda? How to achieve it?

Hi, We give it a try on JetPack 6.1 with a similar script as topic 199891. It can build Open MPI with CUDA support without issue. Could you give it a try? build_openMPI.sh (986 Bytes) Thanks.

Thanks.