search

Kyubyong / dc_tts

A TensorFlow Implementation of DC-TTS: yet another text-to-speech model
Apache License 2.0
1.16k stars 370 forks source link

Running on GPU #74

Open OOps717 opened 4 years ago

OOps717 commented 4 years ago

Hello, I am trying to run training on both my computer and google colab, however even if I am trying to change something it still trains on CPU. I am guessing if there is any solution ?

kheniparth commented 4 years ago

I did face that issue and found that you would have to remove tensorflow package and install tensorflow-gpu package for python and then the code will start utilizing GPU RAM. But my CPU is still getting utilized 100% not sure, why but I can confirm that the training rate is increased exponentially.

Note: tensorflow-gpu package will require some extra configuration to be done or else you will start getting errors, also you would have to you python 3.6 env or else this code breaks (learnt this the hard way )

Pawandeep-prog commented 4 years ago

The solution i am going to share worked for me

first,

unistall tensorflow from environment **pip uninistall tensorflow**

then,

install tensorflow-gpu 1.14 **pip install tensorflow-gpu==1.14**