Error : Failed to get convolutio
2020-12-09 本文已影响0人
h小吴y
Error : Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
https://github.com/tensorflow/tensorflow/issues/24828
image
# 添加下面代码即可
from tensorflow.compat.v1import ConfigProto
from tensorflow.compat.v1import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth =True
session = InteractiveSession(config=config)
