“Failed to get convolution algorithm. This is probably because cuDNN failed to initialize”错误的解决办法
最近在使用TF2.0。运行程序出现以下错误。
Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
一开始怀疑是CUDA和CuDNN配置错误(要求版本匹配)。反复试验后,还是有这个错误。
最后发现可能是GPU内存不足造成的。需要在程序前加以下一段代码:
from tenso...
原创
2019-11-06 17:09:32 ·
56291 阅读 ·
119 评论