【Tensorflow】Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

在训练一个LSTM网络的时候报了这个错

【Tensorflow】Failed to get convolution algorithm. This is probably because cuDNN failed to initialize_第1张图片

 看提示说可能是cuDNN初始化失败。

一开始想到可能是cuDNN的版本问题,但是几经周折,查阅了官方的文档之后,确认cuda和cudnn的版本都是对应的。

然后网上看到说可能是因为显存爆掉了。

果然,我在代码开头添加了

import os
os.environ['CUDA_VISIBLE_DEVICES'] = "-1"

强制使用CPU进行训练。

问题就解决了。

你可能感兴趣的:(深度学习,tensorflow,tensorflow,cuda,cudnn,显存,LSTM)