How to run the code on GPU?

thunlp / OpenNE

An Open-Source Package for Network Embedding (NE)

MIT License

1.68k stars 485 forks source link

How to run the code on GPU? #51

Closed xxxiaol closed 6 years ago

xxxiaol commented 6 years ago

I simply use CUDA_VISIBLE_DEVICES to restrict it on GPU, but it does not work. I'm new to GPU, could you please tell me how to run the code on GPU? Thanks!

FanGhost commented 6 years ago

try installing tensorflow-gpu