Skip to main content
 首页 » 编程设计

使用tensorflow时Python内核死了

2024年10月01日25jirigala

我在 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})  

什么原因可能导致这个问题?

请您参考如下方法:

我不确定 mnist 是否真的成立,但看起来您正在插入整个测试集进行评估,即 10 000 张图像。即使对于中等规模的神经网络,它也很多。该进程很可能因 OOM 而死亡。

尝试传递一个较小的批处理,比如大小为 100。