running on GPU

Question

island255 opened this issue 4 years ago · 2 comments

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?

Answer 1 · 2021-08-06T08:30:43.000Z

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

Answer 2 · 2021-09-15T14:36:24.000Z

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