Tensroflow training error: CUDNN_STATUS_ALLOC_FAILED

详细错误信息:

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

Tensroflow training error: CUDNN_STATUS_ALLOC_FAILED_第1张图片

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

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

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

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

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

扫描下面二维码即可关注

Tensroflow training error: CUDNN_STATUS_ALLOC_FAILED_第2张图片

你可能感兴趣的:(Tensorflow,GPU,深度学习)