How to improve GPU usage?

aspuru-guzik-group / ORGANIC

Code repo for optimizing distributions of molecules.

GNU General Public License v3.0

128 stars 63 forks source link

How to improve GPU usage? #3

Open toushi68 opened 6 years ago

toushi68 commented 6 years ago

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

xuzhang5788 commented 6 years ago

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