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

1365 主题	1856 帖子	1万积分

管理员

Rank: 10

积分: 14439

电梯直达

楼主

发表于 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的显存占用即可

分享到: QQ好友和群

收藏0 转播

让天下人人学会人工智能！人工智能的前景一片大好！

使用道具举报

返回列表

|Archiver|手机版|小黑屋|人工智能工程师的摇篮 ( 湘ICP备2020019608号-1 )

GMT+8, 2024-5-18 07:44 , Processed in 0.202640 second(s), 19 queries .

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