Hi everyone,
I’ve already read few posts about this, but i can’t seem to find an answer.
My problem is that Deepspeech doesn’t seem to run on GPU when training a model. Maybe I’m missing something, but I think I did everything accordingly to the Readme of the repo. Here is what I did:
- create virtualenv with ‘… -p python3’ and activated it
- cloned the repo and did these steps:
pip3 install -r requirements.txt
pip3 install $(python3 util/taskcluster.py --decoder)
pip3 uninstall tensorflow
pip3 install 'tensorflow-gpu==1.14.0'
I also should have all the CUDA dependencys, that’s what my colleague did and also nvidia-smi spits out the following:
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 418.87.00 Driver Version: 418.87.00 CUDA Version: 10.1 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 Quadro P4000 On | 00000000:02:00.0 On | N/A |
| 46% 31C P8 6W / 105W | 383MiB / 8116MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 1173 G /usr/lib/xorg/Xorg 243MiB |
| 0 1391 G /usr/bin/gnome-shell 137MiB |
+-----------------------------------------------------------------------------+
When starting training, Deepspeech gives me some warnings though, one of them is:
WARNING:tensorflow:From /home/encoder80/Desktop/190919_Deepspeech/lib/python3.6/site-packages/tensorflow/python/ops/init_ops.py:1251: calling VarianceScaling.__init__ (from tensorflow.python.ops.init_ops) with dtype is deprecated and will be removed in a future version.
which makes me wonder, shouldn’t it be /site-packages/tensorflow-GPU/ or something? Also pip3 list gives me this output:
absl-py 0.8.0
asn1crypto 0.24.0
astor 0.8.0
attrdict 2.0.1
audioread 2.1.8
bcrypt 3.1.7
beautifulsoup4 4.8.0
bs4 0.0.1
certifi 2019.9.11
cffi 1.12.3
chardet 3.0.4
cryptography 2.7
cycler 0.10.0
decorator 4.4.0
deepspeech-gpu 0.5.1
ds-ctcdecoder 0.6.0a5
gast 0.3.2
google-pasta 0.1.7
grpcio 1.23.0
h5py 2.10.0
idna 2.8
joblib 0.13.2
Keras-Applications 1.0.8
Keras-Preprocessing 1.1.0
kiwisolver 1.1.0
librosa 0.7.0
llvmlite 0.29.0
Markdown 3.1.1
matplotlib 3.1.1
numba 0.45.1
numpy 1.15.4
pandas 0.25.1
paramiko 2.6.0
pip 19.2.3
progressbar2 3.46.1
protobuf 3.9.1
pycparser 2.19
PyNaCl 1.3.0
pyparsing 2.4.2
python-dateutil 2.8.0
python-utils 2.3.0
pytz 2019.2
pyxdg 0.26
requests 2.22.0
resampy 0.2.2
scikit-learn 0.21.3
scipy 1.3.1
setuptools 41.2.0
six 1.12.0
SoundFile 0.10.2
soupsieve 1.9.3
sox 1.3.7
tensorboard 1.14.0
tensorflow-estimator 1.14.0
tensorflow-gpu 1.14.0
termcolor 1.1.0
urllib3 1.25.3
Werkzeug 0.15.6
wheel 0.33.6
wrapt 1.11.2
What am I missing? Hoping that someone can give me a hint or something.
Thanks in advance!
gneulyn
P.s.: everything seems to works fine, just not on the GPU