Pytorch error on multi-GPU setup: cannot import name 'invoke_remote_python_udf' from 'torch.distributed'

Question

I'm getting this below error when importing Pytorch.

"cannot import name 'invoke_remote_python_udf' from 'torch.distributed'"

DukeLover · Accepted Answer · 2019-11-21 13:40:32Z

2

I faced the similar problem. For me , the problem was occuring due to mismatch in cuda version. I am using conda environment.

In my system the nvcc --version showed the result as 10.0' , while I had mistakenly install the pytorch for cuda version10.1`

I removed the torch and torchvision and re-installed them using the following command:

conda install pytorch torchvision cuda100 -c pytorch.

Installing the package cuda100 solved the problem.

answered Nov 21, 2019 at 13:40

DukeLover

2,4972 gold badges21 silver badges27 bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1