AttributeError: module 'torch.distributed' has no attribute 'init_process_group'

Still getting this error after installing latest pytorch-nightly and other stuff, when I tried to run the imagenet example script.