Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize...

遇到以上问题是因为本人运行的代码需要tensorflow1.9.0,但是我的是1.14.0,查了很多办法都是降级版本,所以后来采用了如下方法,将其放到代码最前面即可解决:

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

猜你喜欢

转载自blog.csdn.net/weixin_45191152/article/details/103715785
今日推荐