1.Train
1.1 内存分配问题
报错信息
Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
解决方法
gpu_options = tf.compat.v1.GPUOptions(per_process_gpu_memory_fraction=1.0)
config = tf.compat.v1.ConfigProto(gpu_options=gpu_options)
config.gpu_options.allow_growth = True
session = tf.compat.v1.Session(config=config)