我正在尝试使用Keras训练MNIST数据来制作2D卷积神经网络。我使用Anaconda，python 3.5.0,64bit。刚刚安装了GPU版本的tensorflow并运行
from tensorflow.python.client import device_lib
device_lib.list_local_devices()
获取[name: "/cpu:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 319475955830947664, name: "/gpu:0"
device_type: "GPU"
memory_limit: 3282324684
locality {
bus_id: 1
}
incarnation: 17748449796282282037
physical_device_desc: "device: 0, name: GeForce GTX 960M, pci bus id: 0000:01:00.0"]
它成功地训练了一个简单的完全连接的神经网络，但只要我尝试适合CNN，内核就会死掉并重新启动。使用CPU时不会出现此问题。