【深度学习】问题解决: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
上一篇: 二分图的最大匹配