search

Rayhane-mamah / Tacotron-2

DeepMind's Tacotron-2 Tensorflow implementation
MIT License
2.27k stars 905 forks source link

seems gpu not been used #271

Closed pandaGst closed 2 years ago

pandaGst commented 6 years ago

I run python3 train.py --model='Tacotron' with default parameters in hparams.py. But when I check GPU using with nvidia-smi, Gpu-Util is 0%. Is there any one got same question? Thank you very much!

sidhantls commented 6 years ago

Check out how to find the connected gpu devices on tensorflow- https://stackoverflow.com/questions/38009682/how-to-tell-if-tensorflow-is-using-gpu-acceleration-from-inside-python-shell.

If the gpu device is not in there, make sure you have tensorflow-gpu installed

pandaGst commented 6 years ago

@sid-sundrani Thank you very much! This issue troubled me few days. I will try as your suggestion!

pandaGst commented 6 years ago

I changed some package's verson of my environment: tf-gpu==1.10 cuda==9.0 cudnn==7.1 anaconda==3 python==3.6.7

other packages version is same as requirment.txt gave.

Then it worked. But the capacity of gpu is not hight, just about 20%.

iamjanvijay commented 5 years ago

Thanks. @pandaGst It worked for me as well now. I agree GPU utilization is really low.

pandaGst commented 2 years ago

This issue is inactive for a long time, close now. And reopen it if necessary.