aspuru-guzik-group/ORGANIC

How to improve GPU usage?

Opened this issue · 1 comments

Hi,

This code somehow can only take ~20% GPU usage.
Noticed there are several spots in the code that
"with tf.device("/cpu:0"):"
Modify this condition (delete or replace it to "/gpu0") doesn't effect the speed.
Any further optimizations?

Thank you in advance!
Toushi68

I installed tensorflow-gpu, but it ran still using CPU. I don't know how to change it to using GPU at the beginning. Thanks