RuntimeError: Error while calling cudnnCreate A call to cuDNN failed

东方耀 · 发表于 2020-3-26 10:50:56

报错：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:39

感谢楼主的分享

		自动登录	找回密码
密码			立即注册

[课堂笔记] RuntimeError: Error while calling cudnnCreate A call to cuDNN failed