每当CNN模型被训练时,python总是崩溃
我真的在寻求你的帮助 我有GTX-1070,它是8vram。 我为CUDA9.0下载了tensorflow gpu、CUDA9.0和CUDNN7.0 DNN一切正常。GPU也运行良好 但每当我尝试训练任何与图像有关的模型时,它就会崩溃。 目前,我与keras合作,接受过VGG16的预培训 我尝试使用较小的批量大小,将图像大小调整为64x64。 当我观察这个过程时,GPU的使用率为0%,然后峰值达到100%,然后崩溃。 Spyder说“内核死机,重新启动” 是gtx-1070真的那么缺乏内存还是我遗漏了什么每当CNN模型被训练时,python总是崩溃,python,tensorflow,deep-learning,keras,convolutional-neural-network,Python,Tensorflow,Deep Learning,Keras,Convolutional Neural Network,我真的在寻求你的帮助 我有GTX-1070,它是8vram。 我为CUDA9.0下载了tensorflow gpu、CUDA9.0和CUDNN7.0 DNN一切正常。GPU也运行良好 但每当我尝试训练任何与图像有关的模型时,它就会崩溃。 目前,我与keras合作,接受过VGG16的预培训 我尝试使用较小的批量大小,将图像大小调整为64x64。 当我观察这个过程时,GPU的使用率为0%,然后峰值达到100%,然后崩溃。 Spyder说“内核死机,重新启动” 是gtx-1070真的那么缺乏内存还是我
感谢阅读我想尝试的第一件事是下载Cudnn 7.1 <>这是很好的说明,你可以考虑重新安装CUDA 9。我也曾经做过同样的事情,这让我很沮丧,但自从我做对了以后就没有什么问题了
我以前也遇到过类似的崩溃问题。原因是我的cudnn7.1和tensorflow gpu(使用cudnn7.05预编译)版本不匹配。只要小心,就不会再有问题。“它崩溃了”什么也没说。您应该发布与崩溃相关的实际错误消息。我认为这是Spyder的问题,而不是GPU的问题。试着独立运行python文件。我确实说过错误消息是内核死机了。这就是我能得到的。我不知道它出了什么问题。它在jupyter笔记本中也会以同样的方式崩溃。不过我会试试python文件