Failed to get convolution algorithm. This is probably because cuDNN failed to initialize解决办法

旗真午马爷

于 2019-08-05 17:49:08 发布

阅读量2w

点赞数 27

文章标签：解决办法 cuddn tensorflow keras

本文链接：https://blog.csdn.net/qq_41868689/article/details/98503069

版权

本文介绍如何通过配置TensorFlow和Keras的GPU选项来解决显存溢出问题，适用于cuDNN7.6、CUDA10.0和Python3.6环境下，使用2080Ti显卡的场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

试过改变cuddn版本，但是没啥效果。感觉应该是显存的问题，如下修改：

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config=config) as session:

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
keras.backend.tensorflow_backend.set_session(tf.Session(config=config))

我的环境是cuddn7.6 + cuda10.0 + python3.6 2080ti这样改就可以运行了