Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
解决办法:在开头加入以下代码
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.5)
config = tf.ConfigProto(gpu_options=gpu_options)
config.gpu_options.allow_growth = True
session = tf.Session(config=config)