详细错误信息：

E T:\src\github\tensorflow\tensorflow\stream_executor\cuda\cuda_dnn.cc:352] Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

网上大部分是说显存问题，当然我的GPU是6G 1660Ti确实一般，给出的解决方法就是分配GPU

一般情况上面的方法有用，尴尬的是我是用tensorflow object_detection api来训练的，找了半天没找到哪里设置添加这个代码。。。

最后偶然发现一个帖子说要升级驱动，我用的是tensorflow1.10.0, cuda 9.0, 然后升级又装了一下9.2，重新训练，它会提示我的

显存小可能会导致性能表现不佳，但是可以训练了，训练速度和我1070的gpu速度差不多，此方法解决了我的问题，大家遇到这

个错误可以试试以上两种方法，希望对你有所帮助！

关注【OpenCV与AI深度学习】获得更多学习资讯

扫描下面二维码即可关注