I am using 2.0.0.nv23.05 on my Jetson Orin Nano. I am working on data parallel with pyTorch, but I got this error: from torch.distributed import init_process_group, destroy_process_group ImportError: cannot import name ‘init_process_group’ from ‘torch.distributed’ Is the pyTorch for Jetson not t…

[image] hlau2: But, it still returns False on torch.distributed.is_available(). @hlau2 I haven’t used distributed mode, but I would check torch.__config__.show() and the PyTorch source to see what torch.distributed.is_available() is checking for.

pyTorch 2.0.0.nv23.05

Autonomous Machines Jetson & Embedded Systems Jetson Orin Nano

dusty_nv November 22, 2023, 7:57pm 2

Hi @hlau2, that PyTorch wheel for Jetson wasn’t built with USE_DISTRIBUTED enabled, so it doesn’t have torch.distributed available. You can either disable the code, or rebuild PyTorch with USE_DISTRIBUTED.

You can find instructions on building PyTorch from source in this topic: