使用 DynamicEdgeConv 时出现导入错误

Question

我正在尝试将 iPython 笔记本从 Google 协作实验室移动到渐变笔记本。

在这个过程中我遇到了以下错误：

ImportError: `DynamicEdgeConv` requires `torch-cluster`.

该方法属于 torch_geometric 模块。 浏览the documentation后，我发现了以下代码：

try:
    from torch_cluster import knn
except ImportError:
    knn = None

其中与

if knn is None:
    raise ImportError('`DynamicEdgeConv` requires `torch-cluster`.')

在文档的后面给了我们这个错误。

我想知道你如何解决这个问题。我已经在代码的开头下载并导入了这个 torch_cluster 模块，它没有在那里给出任何错误，所以应该正确安装。