“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 ·
56044 阅读 ·
119 评论