当我在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笔记本......
解决这个问题有什么好主意吗?