本文为博主原创文章,未经博主允许不得转载!
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.
解决办法: 在代码开头导入下面几行代码
import tensorflow
as tf
config
= tf
.ConfigProto()
config
.gpu_options
.allow_growth
= True
sess
= tf
.Session(config
=config
)