search

thunlp / OpenNE

An Open-Source Package for Network Embedding (NE)
MIT License
1.68k stars 485 forks source link

running on GPU #116

Closed island255 closed 3 years ago

island255 commented 3 years ago

Hi, I have a question about running OpenNE on GPU.

I have created a docker with TensorFlow-gpu and can successfully use GPU to run simple codes, but failed to run TADW using GPU.

Is there a way to run this TensorFlow version on GPU or we can only run the PyTorch version on GPU?

island255 commented 3 years ago

It seems that TADW in the PyTorch version is also not implemented to run on GPU. Will it support it in the future?

Bznkxs commented 3 years ago

Hi @island255 ! Thanks for your suggestion. We have added GPU support for TADW in the latest commit.