search

xwhan / DeepPath

code and docs for my EMNLP paper "DeepPath: A Reinforcement Learning Method for Knowledge Graph Reasoning"
533 stars 135 forks source link

Training is very slow #2

Closed MichaelXin closed 6 years ago

MichaelXin commented 6 years ago

Hi Wenhan, I try to train this program, but the training is very slow on my machine. It seems that the program does not use the GPU on my machine. But I'm sure that the tensorflow in my system can employ the GPU acceleration, because I have tested it using other programs. Is there any idea for that? Thank you very much!

xwhan commented 6 years ago

Hi Michael, the code will automatically use GPU if you have tensorflow-gpu installed. Maybe you will need to double check that. Also, you may want you may want check whether the code uses your GPU memory during training.