多GPU设置上的Pytorch错误：无法从“ torch.distributed”导入名称“ invoke_remote_python_udf”

Question

导入Pytorch时出现以下错误。

“无法从“ torch.distributed”中导入名称“ invoke_remote_python_udf””

Answer 1

我遇到了类似的问题。对我来说，问题是由于cuda版本中的不匹配而引起的。我正在使用conda环境。

在我的系统中，nvcc --version将结果显示为10.0' , while I had mistakenly install the pytorch for cuda version 10.1`

我删除了torch和torchvision，然后使用以下命令重新安装它们：

conda install pytorch torchvision cuda100 -c pytorch。

安装软件包cuda100解决了该问题。