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

Posted 2023-03-16

【中文标题】如何修复 RuntimeError：dlpack 不支持 Bool 类型

【英文标题】：How to fix RuntimeError: Bool type is not supported by dlpack

【发布时间】：2021-12-24 11:21:23

【问题描述】：

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

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

谢谢。

【问题讨论】：

请以文本形式提供您的代码！

【参考方案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()

【讨论】：

安装最新版本的 dgl 也很好用 !pip install dgl -f https://data.dgl.ai/wheels/repo.html

我尝试了 0.6.1 版并得到了同样的错误。哪个版本适合您？

【参考方案2】：

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

【讨论】：

您的答案可以通过额外的支持信息得到改进。请edit 添加更多详细信息，例如引用或文档，以便其他人可以确认您的答案是正确的。你可以找到更多关于如何写好答案的信息in the help center。