美文网首页工作生活
tensorflow-gpu深度学习时kernel died r

tensorflow-gpu深度学习时kernel died r

作者: tianchen627 | 来源:发表于2019-06-29 15:28 被阅读0次

没有弹出任何报错信息遇到这种情况,目前我发现的可能的原因有两个:

1.tensorflow-gpu,CUDA,cuDNN版本不匹配。比如:


7.6.0
7.0.5

我配置的环境是python3.6+CUDA9.0+tensorflow-gpu1.7,用cuDNN7.6.0就会出现问题,重新安装cuDNN7.0.5就正常运行了。

2.有其他占大量显存的进程存在,如jupyter notebook。任务管理器和nvidia-smi都可以查看,实在不行重启就行了。

相关文章

网友评论

    本文标题:tensorflow-gpu深度学习时kernel died r

    本文链接:https://www.haomeiwen.com/subject/euspcctx.html