欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【深度学习】问题解决:Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

程序员文章站 2024-03-16 23:44:52
...

目前本人TensorFlow 1.12

1.在import tensorflow as tf 语句后面,设置allow_growth:

config = tf.ConfigProto()
config.gpu_options.allow_growth = True 

2.把原来的sess = tf.Session(不管里面有什么)语句换成:
sess = tf.Session(config = config)

3.重新运行,问题解决

PS:另外一个很麻烦的解决方案是使用以下方法将TensorFlow降级到1.8.0:

pip install --upgrade tensorflowgpu==1.8.0