Internal: cuDNN launch failure : input shape ([1024,32,1,1]) 全网解决方案整理集合

做一个深度学习推荐模型DeepFM的时候出现了这样的错误。

网络搭建问题请看我的csdn_DeepFM实战篇完全版教程—全资源免费下载

解决方案：

• 网上说在开头加上这个，好像没用，加了这句显存反而跑不上去

import os
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'

• 减少batch_size，可以尝试，这边试了没用
  
• 还有个博主说通过删减BN层，链接

这个确实是解决了

注释之后解决。。。。。。不过还得找找更好的解决方案，这样删减已经破坏了原有的网络结构。

• 貌似在bn层加入fused = True可以解决该问题，原因未知，链接