README.md

January 23, 2021 ยท View on GitHub

Ragdoll

Ragdoll is a distributed GNN system on GPUs. Currently, it uses DGL for single GPU computation and GCCL for graph communication.

Prerequisites

git clone --recursively

GCCL installation on ${GCCL_HOME}

python==3.6

DGL==0.4.3.post2

pip install dgl-cu102==0.4.3.post2

Pytorch==1.7.1

pip install torch torchvision

BUILD

./build.sh

Example

GCN

./run_gcn.sh

GIN

./run_gin.sh

Contents

  1. 1Ragdoll
  2. 2Prerequisites
  3. 2.1git clone --recursively
  4. 2.2GCCL installation on ${GCCL_HOME}
  5. 2.3python==3.6
  6. 2.4DGL==0.4.3.post2
  7. 2.5Pytorch==1.7.1
  8. 3BUILD
  9. 4Example
  10. 4.1GCN
  11. 4.2GIN