python 每当训练 CNN 模型时都会崩溃
python keeps crashing whenever CNN model is being trained
我真的很想得到你的帮助。
我有 GTX-1070，它是 8vram。
我为cuda 9.0下载了tensorflow-gpu、cuda 9.0、cudnn 7.0。
DNN 一切正常。 GPU 也工作正常。
但是每当我尝试训练任何与图像有关的模型时，它都会崩溃。
目前我正在使用 keras 预训练的 VGG16。
我尝试使用较小的批量大小，将图像调整为 64x64。
当我查看该过程时，GPU 使用率为 0%，然后峰值达到 100%，然后崩溃。
Spyder 说 "kernel died, restarting".
是 gtx-1070 真的那么内存不足还是我遗漏了什么？
感谢阅读
我要尝试的第一件事是下载 Cudnn 7.1。
这些是很好的指导，您可以考虑重新安装 Cuda 9。我不得不在某一时刻做同样的事情，这令人沮丧，但自从我做对后就没有遇到任何问题。
我之前遇到过类似的崩溃问题。原因是我的cudnn7.1和tensorflow-gpu（用cudnn7.05预编译）版本不匹配。保养一下就没问题了
我真的很想得到你的帮助。
我有 GTX-1070，它是 8vram。 我为cuda 9.0下载了tensorflow-gpu、cuda 9.0、cudnn 7.0。
DNN 一切正常。 GPU 也工作正常。
但是每当我尝试训练任何与图像有关的模型时，它都会崩溃。 目前我正在使用 keras 预训练的 VGG16。
我尝试使用较小的批量大小，将图像调整为 64x64。 当我查看该过程时，GPU 使用率为 0%，然后峰值达到 100%，然后崩溃。 Spyder 说 "kernel died, restarting".
是 gtx-1070 真的那么内存不足还是我遗漏了什么？
感谢阅读
我要尝试的第一件事是下载 Cudnn 7.1。
这些是很好的指导，您可以考虑重新安装 Cuda 9。我不得不在某一时刻做同样的事情，这令人沮丧，但自从我做对后就没有遇到任何问题。
我之前遇到过类似的崩溃问题。原因是我的cudnn7.1和tensorflow-gpu（用cudnn7.05预编译）版本不匹配。保养一下就没问题了