multi-GPU - Githubissues

thunlp / OpenNE

An Open-Source Package for Network Embedding (NE)

MIT License

1.68k stars 485 forks source link

Closed Bznkxs closed 3 years ago

Bznkxs commented 4 years ago

Supports GPU; Modifies model structure:

change procedure of getting self.embeddings and self.vectors;
adds interfaces to register non-pytorch buffers (behaviour may vary between PyTorch versions before 1.5.0 and after);
updates buffers and parameters in specific models; Supports splitting dataset in graph classes; Supports silent mode.

LynnZhang9 commented 3 years ago

Thank you for your work! But I still don't find how can we run this packages on gpu, where should we adjust the script?

Bznkxs commented 3 years ago

Thank you for your work! But I still don't find how can we run this packages on gpu, where should we adjust the script?

@LynnZhang9 please refer to readme.md at the pytorch branch(https://github.com/thunlp/OpenNE/tree/pytorch). It should be straightforward to run the stable branch with GPU.