Is it only for CPU training? about tacotron HOT 5 CLOSED

You need to install the tensorflow-gpu to do so

Check it out on Tensor Flow Installation

from tacotron.

@marcossilva
Thank you for your prompt answer. I already have installed Tensorflow for GPU. I've been training with GPU since months ago but for the code released, so far, I haven't figured out why it does not use my GPU.

When typing nvidia-smi I can see the projects that are running on the GPU and this code does not run on the GPU. Do you see that your code is running in the GPU when typing nvidia-smi?

from tacotron.

@basuam I haven't checked with the nvidia-smi because when the training or the evaluation starts it shows that it allocated my GPU0 (my only GPU in fact) to the training. I double checked here just to be sure and it shows python3 in the nvidia-smi processes. Are you using a virtualenv to make sure the requirements are correct and thath the tensorflow package isn't installed, just the tensorflow-gpu?

I've just re-read your post and realized it tries to allocate the GPU but it can't. It might be something else in that case. Maybe the GPU is allocated with another project?

from tacotron.

@marcossilva
I'm running with Python 2.7.5 and I do not see that message that my GPU has been allocated.
Maybe that's the problem. Let me try with Python 3.
Thank you for your help.

UPDATE:
@marcossilva. You were right, certainly, when using Python 3 the GPUs are assigned correctly.
I guess Python3 needs to be added to the requirements section ( @Kyubyong )

from tacotron.

Closing this issue because now it is clear that Python3 is needed for the GPU capability.

from tacotron.