ImportError: cannot import name 'Store' from 'torch.distributed'

[janelin_312]

I use python 3.8.10, torch 2.0.0+nv23.5;

• Jetson AGX Orin 64GB
  Jetpack 5.1

import torch
torch.version
‘2.0.0+nv23.05’
from torch.distributed import Store
Traceback (most recent call last):
File “”, line 1, in
ImportError: cannot import name ‘Store’ from ‘torch.distributed’ (/home/jane/.local/lib/python3.8/site-packages/torch/distributed/init.py)
torch.distributed.is_available()
False
`

does it mean that the distributed package does not work on Orin?
What should I do to to make ‘from torch.distributed import Store’ work well?

[dusty_nv]

@janelin_312 it’s not built with distributed enabled (it’s not a common use-case for Jetson) - you can rebuild PyTorch with USE_DISTRIBUTED=1 (after installing libopenmpi-dev). See the “Build from Source” part of this thread: PyTorch for Jetson