I’m doing some tests for CPU and GPU environment usages for prediction (Predict.py). I’m using an audio file Audio: mp3, 44100 Hz, stereo, fltp, 192 kb/s of duration 00:03:15.29 On a Intel i7 - 12 core CPU the prediction time log says Completed after 0:03:19 while on Intel Xeon 12 core plus 2x Nvidia GeForce GTX 1080 says Completed after 0:00:16 I’m not sure from logging if tensorflow is using both gpu devices or gpu 0 only. If I’m not wrong, most of the work is done in the Models.py here https://github.com/f90/Wave-U-Net/blob/master/Models/UnetSpectrogramSeparator.py#L39 when the computation graph is calculated. I assume that these operations go on gpu:0 in this configuration, so gpu:1 will not be used - but I’m not sure of it. Thank you very much! |