本文为博主原创文章,未经博主允许不得转载!
ubuntu18.04 cuda10.0+cudnn7.5+tensorflow1.13.1
报错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.
解决办法:
在代码开头导入下面几行代码
// An highlighted block
import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)