如何修复运行时错误：dlpack 不支持 Bool 类型

Question

我一直使用下面的代码来获取DGL提供的默认Cora数据集，但是今天突然出现了以下错误。

该代码在 CoLab（python 3.7 和 Pytorch 后端）中运行。我相信这是 DGL 更新中的错误（因为它之前一直有效）。但是，我只是想知道我们是否可以做些什么来解决这个问题？

谢谢。

Answer 1

看来错误是从 torch 更新到 1.10.0 时出现的。将 torch 重新安装到 1.9.1 对我有用。您可以在colab中重新安装torch，如下所示：

!pip install dgl==0.6.1
!pip install torch==1.9.1
import dgl
cora = dgl.data.CoraGraphDataset()