Jupyter笔记本内核死了(内存不足)

时间:2016-10-13 11:27:52

标签: memory anaconda jupyter keras

当我在Jupyter笔记本中使用keras模型(深度学习库)时遇到问题。 我在conda中安装scipy,matplotlib库后发生此问题。 在执行代码单元之前,我的计算机GPU内存使用率为241 MB / 8113 MB

我执行了简单的keras模型代码,如下所示

from keras.models import Sequential
from keras.layers import Convolution1D, MaxPooling1D, Activation, Dense, Flatten

# Parameter
n_filters1 = 64 # number of convolutional filters
n_filters2 = 32
n_conv = 4    # convolution filter size
n_pool = 3    # pooling window size

model = Sequential()
model.add(Convolution1D(n_filters1, n_conv, border_mode='valid', activation='relu', input_shape=(40, 1)))
model.add(MaxPooling1D(n_pool, stride=2))
model.add(Convolution1D(n_filters2, n_conv, border_mode='valid', activation='relu'))
model.add(MaxPooling1D(n_pool, stride=2))

model.add(Flatten())
model.add(Dense(100, activation='relu'))
model.add(Dense(2, activation='softmax'))

执行上述代码后,GPU内存使用量从241MB变为 - > 7754MB

所以,当我运行训练细胞代码时,Jupyter死了

这个问题一次又一次地发生......即使我重新启动了我的Jupyter笔记本......

解决这个问题有什么好主意吗?

0 个答案:

没有答案