cugraph扩展用于DGL
项目描述
cugraph_dgl
描述
RAPIDS cugraph_dgl提供了一种基于cugraph的DGLGraph类的鸭子类型版本,该版本用于存储图结构和节点/边特征数据。使用cugraph作为后端,允许DGL用户访问一系列用于图分析的GPU加速算法,如中心性计算和社区检测。
Conda
使用命令安装和更新cugraph-dgl及所需依赖项
conda install mamba -n base -c conda-forge
mamba install cugraph-dgl -c rapidsai-nightly -c rapidsai -c pytorch -c conda-forge -c nvidia -c dglteam
从源码构建
创建conda开发环境
mamba env create -n cugraph_dgl_dev --file conda/cugraph_dgl_dev_11.6.yml
以可编辑模式安装
pip install -e .
运行测试
pytest tests/*
用法
+from cugraph_dgl.convert import cugraph_storage_from_heterograph
+cugraph_g = cugraph_storage_from_heterograph(dgl_g)
sampler = dgl.dataloading.NeighborSampler(
[15, 10, 5], prefetch_node_feats=['feat'], prefetch_labels=['label'])
train_dataloader = dgl.dataloading.DataLoader(
- dgl_g,
+ cugraph_g,
train_idx,
sampler,
device=device,
batch_size=1024,
shuffle=True,
drop_last=False,
num_workers=0)
项目详情
关闭
cugraph_dgl_cu11-24.8.0.tar.gz的哈希值
算法 | 哈希摘要 | |
---|---|---|
SHA256 | 288ab0b37c3c2897d0f206ee4d1d6f373d996e8a42e45cdec0892c5463b5f5d9 |
|
MD5 | f103f17e771bc056db600e5adfe7ed9f |
|
BLAKE2b-256 | 9af7f995faecc1ccb0b97966b7f51974287429c3c5a356fcea38d06dcb361a97 |