Failed to get convolution algorithm. This is probably because cuDNN failed to initialize解决方法
如果tensorflow-gpu的版本是2.0.0,那直接用下面这5句代码就可以解决
from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)
原因可能是:gpu内存不够,不是版本不匹配
注意:这种解决方法只限于tensorflow-gpu的版本是2.0.0,同样在tensorflow-gpu的版本是2.3.1的情况下测试了一下,这种方法不可行。
如果是2.3.1版本
可以用下面方式
import os
os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘/device:GPU:0’