无法在GPU上的嵌入层顶部添加LSTM层 - 具有张量流后端的Keras

Question

以下是代码段：

from keras.models import Sequential
from keras.layers import LSTM, Embedding

lang_model = Sequential()
lang_model.add(Embedding(1000, 100, input_length=25))
lang_model.add(LSTM(100,return_sequences=True)) #stuck here

lang_model.summary()

当我在我的本地CPU上运行上面的代码时，它运行正常，但是当在谷歌云上的GPU上运行时，它只是不起作用。它甚至没有显示任何错误，它只是卡在第3行。

请建议。

刚才发现它与theano后端工作正常，但是当我使用tensorflow后端时失败了。

感谢。