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

Question

我真的在寻求你的帮助。

我有 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 真的是内存不足还是我遗漏了什么？

谢谢阅读

Answer 1

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

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

安装说明

Answer 2

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