在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上運行時，為什么此腳本會失敗，並顯示“分段錯誤（核心已轉儲）”？ 在CPU上正常工作。

Answer 1

如果在GPU上使用Keras 2運行K.clearsession（），則可能會遇到分段錯誤。 如果您的代碼中有此內容，請嘗試將其刪除！ 祝好運！