在GPU上运行时Keras中的分段错误

Question

这是我的Keras模型：

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

model = Sequential()
model.add(Embedding(max_features, output_dim=256))
model.add(LSTM(128))
model.add(Dropout(0.5))
model.add(Dense(1, activation='sigmoid'))

环境：keras 2.1，python 3.6，tensorflow后端。为什么这个脚本在GPU上运行时会出现＆＃34; Segmentation fault（core dumped）＆＃34;？在CPU上运行没有错误。

Answer 1

如果在GPU上使用Keras 2运行K.clearsession（），则可能会遇到分段错误。如果您的代码中有此内容，请尝试将其删除！祝你好运！