我在Windows 7操作系统中使用Anaconda 4.3.1(64位)下的Python 3.5.2运行仅CPU版本的TensorFlow。当我运行以下代码时,Python内核崩溃且无法重启,显示了一些错误但没有错误信息提示。
print ("test accuracy %.3f" % accuracy.eval(feed_dict={ x: mnist.test.images, y_: mnist.test.labels, keep_prob: 1.0}))
但是当我运行类似的代码时,Python正常工作:
train_accuracy = accuracy.eval(feed_dict={ x:batch[0], y_: batch[1], keep_prob: 1.0})
导致这个问题的可能原因是什么?
回答:
我不确定mnist
具体包含什么内容,但看起来你是在一次性评估整个测试集,总共有10,000张图片。即使对于中等大小的神经网络来说,这也是一个很大的数据量。进程很可能因为内存溢出(OOM)而崩溃。
尝试传递一个较小的批次,比如大小为100的批次。