工作生活

tensorflow-gpu深度学习时kernel died r

2019-06-29  本文已影响0人  tianchen627

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

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都可以查看,实在不行重启就行了。

上一篇 下一篇

猜你喜欢

热点阅读