Training is not using GPU

tensorflow / nmt

TensorFlow Neural Machine Translation Tutorial

Apache License 2.0

6.37k stars 1.96k forks source link

Training is not using GPU #228

Closed denisb411 closed 6 years ago

denisb411 commented 6 years ago

Why is this happening? Training is too slow on CPU:0...

How can I set gpu:0 ?

oahziur commented 6 years ago

IteratorGetNext is part of your input pipeline, so it is running on CPU. You can get detail device placement information by setting --log_device_placement=True.

denisb411 commented 6 years ago

So the training is indeed made on GPU (if avaible)?

oahziur commented 6 years ago

@denisb411 you can check GPU utilization to confirm it is using GPU. You can also set --num_gpus=0 compare the speed with CPU only training.

lmthang commented 6 years ago

Close for now due to no further activity.