Keras在运行我的玩具示例时内存不足

from keras.models import Sequential
from keras.layers import Dense

model = Sequential()
model.add(Dense(32, input_dim=784, activation='relu'))  # 减少神经元数量
model.add(Dense(10, activation='softmax'))

model.fit(x_train, y_train, batch_size=32)  # 减小batch_size

from keras.preprocessing.image import ImageDataGenerator

datagen = ImageDataGenerator()
datagen.fit(x_train)
model.fit_generator(datagen.flow(x_train, y_train, batch_size=32), steps_per_epoch=len(x_train) / 32, epochs=epochs)

from keras.utils import multi_gpu_model

# 假设你有两块GPU
parallel_model = multi_gpu_model(model, gpus=2)
parallel_model.compile(loss='categorical_crossentropy', optimizer='adam')
parallel_model.fit(x_train, y_train, batch_size=32)

pip install --upgrade keras tensorflow