深度学习-常见问题记录

1.tensorflow-gpu训练报错

1.1 “Failed to get convolution algorithm. This is probably because cuDNN failed to initialize”错误的解决办法：
对GPU进行按需分配

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)