我真的在寻求你的帮助。
我的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真的缺少内存还是我错过了什么？
感谢您阅读
答案 0 :(得分：0)
我要尝试的第一件事是下载Cudnn 7.1。
这些是很好的说明，您可以考虑重新安装Cuda 9。我不得不一次做同样的事情，这令人沮丧，但是因为我做对了就没有问题。
答案 1 :(得分：0)
之前我遇到过类似的崩溃问题。原因是我的cudnn7.1和tensorflow-gpu（预编译与cudnn7.05）版本不匹配。一旦注意，就没有问题了。