搞定cuDNN启动失败问题的方法
最编程
2024-08-09 13:55:25
...
- 小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。
在神经网络模型推断过程中,报错
Internal: cuDNN launch failure : input shape ([1,3,2048,2048]) node bn_data/FusedBatchNorm
,本文记录解决方案。
原因分析
网上有怀疑该问题由BN层引起,去掉BN层可以解决问题TensorFlow实战 InternalError: cuDNN launch failure : input shape(未解决求交流)。
更换cudnn版本也无济于事。
事实上更可能是TensorFlow显存配置的问题(事实上也是缺少证据的,只是按照这个思路可以排除故障,因此该因素应该是最大似然的原因)
解决方案
在代码中加入按需分配显存的配置:
方案1
con = tf.ConfigProto()
con.gpu_options.allow_growth = True
keras.backend.tensorflow_backend.set_session(tf.Session(config=con))
方案2
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
原理相同,只是不一样的写法。
上一篇: 聪明的猜拳技巧:避免后悔的反事实算法解析
下一篇: Qt Creator 快捷键速查表
推荐阅读
-
搞定 SSH 连接缓慢问题的方法详述
-
搞定cuDNN启动失败问题的方法
-
搞定List集合去重问题:探究Distinct内置方法的妙用
-
搞定promise中出现的Uncaught reason问题:只需添加.catch()方法即可避免错误
-
搞定!一次MySQL 1054错误的解决方法:Where子句中未知列的问题
-
搞定App内存问题:一步步分析和优化内存泄漏的方法
-
搞定MATLAB R2012a安装及激活码失效问题的方法 - 转载
-
九种搞定Nginx 502 Bad Gateway问题的方法及其背后的原因解析
-
如何订阅onlyfans充值,一文带你了解火了的onlyfans(最新)常见问题:年龄验证失败、支付失败解决方法
-
Win10 添加引导启动项的方法和无效问题解决方案