msr-fiddle/pipedream

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

Closed this issue · 1 comments

Why always appear this error, I have according to the readme: docker pull nvcr.io/nvidia/pytorch:20.03-py3, use docker to run pp2bw train, but still raise the wrong

it should use docker 22.12:
nvidia-docker pull nvcr.io/nvidia/pytorch:22.12-py3