使用tensorflow时Python内核死了

Question

我在Windows 7操作系统中的anaconda 4.3.1（64位）中使用python 3.5.2下的tensorflow（仅CPU版本）。当我运行以下代码时，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}) 

哪个原因可能导致这个问题？

Answer 1

我不确定mnist实际上是否成立，但看起来你正在推动整个测试集进行评估，即10 000张图像。即使对于中等规模的神经网络也是如此。这个过程可能会因OOM而死亡。

尝试传递较小的批次，比如大小为100。