关于tensorflow-gpu的程序用spyder运行后,控制台报错: Kernel died, restarting的解决方法

问题:

安装tensorflow-gpu以及cuda和cudnn以后,运行原先cpu版时可行的代码,spyder的控制台报错 Kernel died, restarting

原因:

我尝试通过 python xxx.py的方式运行代码文件,发现报错如下:

这里圈起来的这句话意思就是我的cudnn是7.1.02版本的,即大版本号是7.1,而cuda源码是用7.0.03编译的,即大版本号是7.0。说白了就是cudnn的版本号不对

解决方法:

去https://developer.nvidia.com/rdp/cudnn-download上下载一个7.0的cudnn((与自己用的cuda版本一致的),然后解压,然后吧bin、lib、include三个文件夹拷贝到C:\ProgramData\NVIDIA GPU Computing Toolkit\v9.0下即可

发布了55 篇原创文章 · 获赞 238 · 访问量 21万+

猜你喜欢

转载自blog.csdn.net/zkp_987/article/details/79763831