Open JangminSon opened 6 years ago
That line of code line 31
is for preprocessing data in optimize.py. It does not affect training, it only loads files into memory.
line 42
with tf.Graph().as_default(), tf.Session() as sess:
is where training occurs, and it should use your GPU (if you have one) by default.
I have question about codes In src/optimize.py,
with tf.Graph().as_default(), tf.device('/cpu:0'), tf.Session() as sess:
I Thought it is meant to training on the CPU (I mean using cpu). And if I run this code, It only use GPU Memory and doesn't use its processing power. and If I trying to change cpu into gpu, It occurs error below.
So I erased the 'tf.device ' part and It worked fine. Am I misunderstood something?
I'm using TF version - 1.4 Python 2.7.12 CUDA 8.0 with cudnn
I hope it would help someone with this error like me.