无法获取卷积算法的问题,提示可能因为cuDNN无法初始化。
打开代码文件在前面添加如下行,问题得以解决。
from tensorflow.compat.v1 import ConfigProto from tensorflow.compat.v1 import InteractiveSession config = ConfigProto() config.gpu_options.allow_growth = True session = InteractiveSession(config=config)