Torch distributed, ProcessGroup import error

AI & Data Science Deep Learning (Training & Inference)
1

I’m hitting some error trying to run some code as I can’t import ‘ProcessGroup’

Python 3.8.10 (default, Jun 22 2022, 20:18:18)
[GCC 9.4.0] on linux
Type “help”, “copyright”, “credits” or “license” for more information.

from torch.distributed import ProcessGroup
Traceback (most recent call last):
File “”, line 1, in
ImportError: cannot import name ‘ProcessGroup’ from ‘torch.distributed’ (/home/fbarilla/.local/lib/python3.8/site-packages/torch/distributed/init.py)
import torch

torch.version
‘1.12.0a0+84d1cb9.nv22.4’

2

Hi,

Looks like you incorrectly posted to the AMA section. Unfortunately there are no technical resources watching this category. Can you please let me know what forum best fits your issue? I will move this topic for you.

3

Thanks Tom, the answer has been provided in: https://forums.developer.nvidia.com/t/torch-distributed-not-supported-on-orin/220732

1 Like
4

Great, I will close this topic then.

Cheers,
Tom