Why don’t you try it out for yourself and see?
In my case, training a model with LSTM
took 10mins 30seconds.
Simply switching the call from LSTM()
to CuDNNLSTM()
took less than a minute.
I also noticed that switching to CuDNNLSTM()
speeds up model.evaluate()
and model.predict()
substantially as well.