dgl torch 不兼容错误:无法加载 Graphbolt C++ 库

问题描述 投票:0回答:1

我正在尝试通过

DGL 安装说明在 google colab 中安装和设置 
dgl

!pip install torch !pip install dgl -f https://data.dgl.ai/wheels/torch-2.3/cu121/repo.html
安装 

pytorch2.3.0+cu121

dgl2.3.0+cu121

但是,当我尝试导入 dgl 时,遇到以下错误:

ImportError: Cannot load Graphbolt C++ library
我也尝试过其他版本的 dgl,但总是遇到某种导入错误。

!nvcc --version

显示cuda版本12.2

问题是否与 cuda 版本不匹配(12.1 v 12.2)?

这个线程这个线程似乎都遇到了类似的问题。

如果有人可以在 Colab 中使用此功能,请分享工作簿链接。

pytorch google-colaboratory importerror dgl
1个回答
0
投票
当我不安装特定于 CPU 的 DGL 版本时导入工作

!pip install dgl -f https://data.dgl.ai/wheels/torch-2.3/repo.html


如果您需要 CUDA,这不是真正的修复,但这意味着 Colab CUDA (12.2) 和 DGL 之间的 CUDA 版本不匹配(看起来只支持最高 12.1)是问题

© www.soinside.com 2019 - 2024. All rights reserved.