search

muhanzhang / SEAL

SEAL (learning from Subgraphs, Embeddings, and Attributes for Link prediction). "M. Zhang, Y. Chen, Link Prediction Based on Graph Neural Networks, NeurIPS 2018 spotlight".
607 stars 142 forks source link

GPU running #37

Closed Amirhekmatshoar closed 3 years ago

Amirhekmatshoar commented 4 years ago

Hello Excuse me, when i run the main.py with USAir, the code running on my CPU How can i run SEAL on GPU I installed CUDA, torch and DGCNN too. thanks

muhanzhang commented 4 years ago

Hi, by default SEAL should run on GPU, unless you specify "--no-cuda". Can you test whether your other programs (such as DGCNN) can run on cuda?

joc32 commented 4 years ago

Try using the PyTorch's Geometric implementation of SEAL, works on GPU (Colab).