import torch
import torch.distributed as dist
print("Is dist available: ", dist.is_available())
print("Pytorch version: ", torch.__version__)
store = dist.HashStore()
# store can be used from other threads
# Use any of the store methods after initialization
store.set("first_key", "first_value")
Output
Is dist available: True
Pytorch version: 2.4.1+cu124
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
Cell In[12], line 7
4 print("Is dist available: ", dist.is_available())
5 print("Pytorch version: ", torch.__version__)
----> 7 store = dist.HashStore()
8 # store can be used from other threads
9 # Use any of the store methods after initialization
10 store.set("first_key", "first_value")
AttributeError: module 'torch.distributed' has no attribute 'HashStore'