Internal: cuDNN launch failure : input shape ([1024,32,1,1]) 全网解决方案整理集合
做一个深度学习推荐模型DeepFM的时候出现了这样的错误。解决方案：网上说在开头加上这个，好像没用，加了这句显存反而跑不上去import osos.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'减少batch_size，可以尝试，这边试了没用还有个博主说通过删减BN层，链接这个确实是解决了注释之后解决。。。。。。不过还得找找更好的解决方案，这样删减已经破
·
做一个深度学习推荐模型DeepFM的时候出现了这样的错误。
网络搭建问题请看我的csdn_DeepFM实战篇完全版教程—全资源免费下载
解决方案：
- 网上说在开头加上这个，好像没用，加了这句显存反而跑不上去
import os
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
- 减少batch_size，可以尝试，这边试了没用
- 还有个博主说通过删减BN层，链接
这个确实是解决了
注释之后解决。。。。。。不过还得找找更好的解决方案，这样删减已经破坏了原有的网络结构。
- 貌似在bn层加入fused = True可以解决该问题，原因未知，链接
一站式 AI 云服务平台
更多推荐
2
0- 0
已为社区贡献2条内容
所有评论(0)