Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED
当你看到这个就证明你的显卡不怎么行了,显存不够
tensorflow的配置加上下面的代码
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)
Hint: If you want to see a list of allocated tensors when OOM happens
出现这个问题也是显卡太渣,改小batch_size