Import 多GPU设置上的Pytorch错误：无法导入名称'；调用远程python udf'；从'；火炬。已分发'；

我在导入Pytorch时遇到以下错误

---

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

我遇到了类似的问题。对我来说，问题的出现是因为

cuda

版本不匹配。我正在使用

conda

环境

在我的系统中，

nvcc--version

显示结果为

10.0'，而我错误地为cuda版本安装了pytorch

10.1`

我移除了

火炬

和

火炬视觉

，并使用以下命令重新安装了它们：

conda安装pytorch torchvision cuda100-c pytorch

安装软件包

cuda100

解决了问题