How to improve GPU usage?
Opened this issue · 1 comments
toushi68 commented
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
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