东方耀AI技术分享

标题: RuntimeError: Error while calling cudnnCreate A call to cuDNN failed [打印本页]

---

作者: 东方耀    时间: 2020-3-26 10:50
标题: RuntimeError: Error while calling cudnnCreate A call to cuDNN failed
报错：RuntimeError: Error while calling cudnnCreate(&handles[new_device_id]) in file /tmp/pip-install-mxk66dc_/dlib/dlib/cuda/cudnn_dlibapi.cpp:104. code: 4, reason: A call to cuDNN failed



程序启动的时候加载了一个keras的模型  后面又调用了dlib的人脸检测，因为keras是默认占用全部显存的，
所以导致后面的调用没有显存了  无法创建cuDNN


控制一下keras的显存占用即可

---

作者: 栀子花    时间: 2020-4-5 00:13

感谢楼主的分享

---

欢迎光临 东方耀AI技术分享 (http://www.ai111.vip/)

Powered by Discuz! X3.4