Google Colab安装DGL遇Graphbolt C++库加载失败问题求助

问题根源 #

你遇到的核心问题是CUDA版本不匹配：Colab当前默认提供的CUDA版本是12.2，但你安装的是适配CUDA12.1的DGL和PyTorch包，导致底层C++库加载失败。

修复步骤 #

1. 卸载现有不兼容的包 #

执行以下命令清理已安装的PyTorch和DGL：

!pip uninstall -y torch dgl

2. 安装适配CUDA12.2的PyTorch和DGL #

• 安装对应CUDA12.2的PyTorch 2.3.0：

!pip install torch==2.3.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu122

• 安装对应PyTorch2.3+CUDA12.2的DGL 2.3.0：

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

3. 验证安装有效性 #

运行以下代码确认环境正常：

import torch
import dgl

# 检查CUDA是否可用
print(f"CUDA available: {torch.cuda.is_available()}")
# 检查DGL版本
print(f"DGL version: {dgl.__version__}")

如果输出CUDA available: True和对应版本号，说明环境配置成功。

可用的Colab工作簿 #

可以直接在Colab中新建笔记本，复制上述步骤运行即可快速搭建可用环境，也可参考社区验证过的配置流程完成环境初始化。

内容的提问来源于stack exchange，提问作者liamhp