常规的报错原因和其他的博客一样。
- 是由于数据量的问题,是数据量n 、batch_size 、GPU数量得满足一定的关系,大家在网上搜一下,基本可以找到。
- tensorflow、tensorflow-gpu 与cuda、cuDNN的版本号不对
本人属于这种问题,然后报错这种,那张纸在纸上算起了基本除法题目,就是为了n/batch_size > gpu数量,纸都戳破了,还是报错,最后我换了项目带同样的模型,又报错了。所以基本可以认定我的环境出了问题,于是我问和我用同一个环境的兄弟,他说他换了tensorflow等等的版本。我换回来了,果断解决了。
附上版本对应的链接:
https://blog.csdn.net/qq_27825451/article/details/89082978