Tacotron-2: seems gpu not been used

Created on 17 Nov 2018  路  4Comments  路  Source: Rayhane-mamah/Tacotron-2

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!

Most helpful comment

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%.

All 4 comments

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

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

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%.

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

Was this page helpful?
0 / 5 - 0 ratings

Related issues

mrgloom picture mrgloom  路  4Comments

mindmapper15 picture mindmapper15  路  10Comments

zhf459 picture zhf459  路  12Comments

zuzana picture zuzana  路  4Comments

sunnnnnnnny picture sunnnnnnnny  路  3Comments