search

microsoft / gated-graph-neural-network-samples

Sample Code for Gated Graph Neural Networks
MIT License
1.03k stars 264 forks source link

Do you use GPU for training? #13

Closed xuzhang5788 closed 6 years ago

xuzhang5788 commented 6 years ago

Thank you for your great work.

When you train your model, do you set your tensorflow with GPU or not? How long will it take for your training?

I pip install -r requirements.txt, then run. It showed that I am using CPU. Even though I install tensowflow-gpu, it still running on CPU. Any advice? Thanks a lot.

xuzhang5788 commented 6 years ago

I solved this problem. Thanks.

mmjb commented 6 years ago

[Closing as indicated]