Skip to content

Latest commit

 

History

History
27 lines (17 loc) · 796 Bytes

README.md

File metadata and controls

27 lines (17 loc) · 796 Bytes
Raw

Distributed Graph Neural Network Learning

Install Metis

download and extract metis:

http://glaros.dtc.umn.edu/gkhome/metis/metis/download
gunzip metis-5.x.y.tar.gz
tar -xvf metis-5.x.y.tar

if you don't have root access, compile metis with prefix

make config prefix=<lib-folder>

Build PyTorch Sparse

make Pytorch Sparse with metis support:

cd pytorch_sparse
export CPATH=/export/local/mfr5226/lib/metis/include/:/usr/local/cuda/include/:$CPATH 
export LD_LIBRARY_PATH=/export/local/mfr5226/lib/metis/lib/:$LD_LIBRARY_PATH 
WITH_METIS=1 LDFLAGS='-L/export/local/mfr5226/lib/metis/lib/' python setup.py build -j 10
python setup.py install

If you have metis in regular location, just compile with: WITH_METIS=1 python setup.py install -j 10