**
GPU显存问题
**
使用服务器跑深度学习代码时出现了这样的问题:
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
import keras
config = tf.compat.v1.ConfigProto(allow_soft_placement=True)
config.gpu_options.per_process_gpu_memory_fraction = 0.3
tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))
具体科学的解释可以参照这里:
keras初始化显存占用问题