python - 如何修复 RuntimeError：dlpack 不支持 Bool 类型

Question

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

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

谢谢。

Answer 1

似乎从 torch 更新到 1.10.0 出现了错误。将手电筒重新安装到 1.9.1 对我有用。您可以在 colab 中重新安装 torch，如下所示：

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

Answer 2

安装最后一个 GPU 加速（cuda 11.1）dgl pip install dgl-cu111 -f https://data.dgl.ai/wheels/repo.html