每当训练 CNN 模型时，python 都会崩溃答案

【问题标题】：python keeps crashing whenever CNN model is being trained每当训练 CNN 模型时，python 都会崩溃
【发布时间】：2018-04-13 10:21:27
【问题描述】：

我真的在寻求你的帮助。

我有 GTX-1070，它是 8vram。我为 cuda 9.0 下载了 tensorflow-gpu、cuda 9.0、cudnn 7.0。

DNN 一切正常。 GPU 也运行良好。

但每当我尝试训练任何与图像有关的模型时，它都会崩溃。目前我正在使用 keras 预训练的 VGG16。

我尝试使用较小的批量大小，将图像大小调整为 64x64。当我查看这个过程时，GPU 的使用率为 0%，然后飙升至 100%，然后崩溃。 Spyder 说“内核死了，正在重新启动”。

gtx-1070 真的是内存不足还是我遗漏了什么？

感谢阅读

【问题讨论】：

【解决方案1】：

我会尝试的第一件事是下载 Cudnn 7.1。

这些是很好的指导，您可以考虑重新安装 Cuda 9。我不得不在某一时刻做同样的事情，这很令人沮丧，但自从我做对了之后就没有问题了。

【讨论】：

【解决方案2】：

我之前也遇到过类似的崩溃问题。原因是我的 cudnn7.1 和 tensorflow-gpu（用 cudnn7.05 预编译）版本不匹配。照顾好以后就没有问题了。

【讨论】：