【最强解法】Failed to get convolution algorithm. This is probably because cuDNN failed to initialize...

网上有说cuDNN的版本问题,我也没有试过这种方法,因为看上去比较复杂耗时,所以我用了别的方法,调整了GPU的利用率(很久之前看的一篇博客的,现在忘记是哪篇了),只需要在代码的前面插入以下代码,就可以轻松解决了,简单有效,我特意写个博客记录下来,同时希望对网友有帮助。

import tensorflow as tf
config = tf.compat.v1.ConfigProto(allow_soft_placement=True)
config.gpu_options.per_process_gpu_memory_fraction = 0.2
tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))

你可能感兴趣的:(Bug处理集锦,bug,深度学习,tensorflow,神经网络)