1、前天在jupyter notebook 上运行tensorflow的代码时(主要是卷积神经网络,用小笔记本上的GPU),运行没多久,出现“The kernel appears to have died. It will restart automatically”。
试了很多遍,一直都是这样。但是又运行之前的Tensorflow基础运算和全连接神经网络的程序,都可以运行,只有开始跑卷积神经网络的时候出现崩溃问题。
在CSDN上浏览了很多博主的经验贴解决方案,但无奈自己试了后还是不行,遂到github上去寻找,于是找到了一个同样问题的讨论。github讨论
其中最高赞答案:
安装了这个包之后,我的tf也无法正常调用了2333(tf1.5,cuda9.0,cudnn7,可能是tf版本不同导致)
后面把这个包删了还是无法调用tf,于是装上了这个包,把它和tf1.5一块卸了,再把tf1.5装上,恢复到了原来状态,可以调用tf了2333.
最后仔细看了下cmd上的错误提示:
2、最终解决方案:将cudnn降级,从7.4降到7.0版本
配置:win7系统
GTX960M
tensorflow-gpu1.5
cuda9.0
cudnn7.4
首先下载cudnn对应版本,我的是v7.0.5(属于7.0版本) cudnn下载
将下载下来cuDNN压缩包解压后的bin、include、lib文件夹
直接复制到cuda目录下:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
文件会自动合并,不用再一个文件夹一个文件夹的放了。
再次运行jupyter notebook 就不会出现Dead kernel 的问题了。
(ps:第一次写博客,希望对同为刚入门的小白有所帮助)
参考:https://blog.csdn.net/weixin_41681293/article/details/82946787
https://github.com/tensorflow/models/issues/3835