Hi,
I’m trying to use distributed autograd. But I’m running into this error.
I have initialized the process group with dist.init_process_group()
with dist_autograd.context() as context_id:
  File "/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/distributed/autograd/__init__.py", line 33, in __enter__
    self.autograd_context = _new_context()
RuntimeError: Need to initialize distributed autograd using torch.distributed.autograd.init()

But I see no init method in torch.distributed.autograd
>>> from torch.distributed import autograd
autog>>> autograd
<module 'torch.distributed.autograd' from '/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/distributed/autograd/__init__.py'>
>>> autograd.
autograd.DistAutogradContext(  autograd.division              autograd.sys
autograd.absolute_import       autograd.get_gradients(        autograd.torch
autograd.backward(             autograd.is_available(         autograd.unicode_literals
autograd.context(              autograd.print_function
>>> autograd.
autograd.DistAutogradContext(  autograd.division              autograd.sys
autograd.absolute_import       autograd.get_gradients(        autograd.torch
autograd.backward(             autograd.is_available(         autograd.unicode_literals