Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

Problem:Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

Solve:

      config = tf.ConfigProto()
      config.gpu_options.allow_growth = True
      session = tf.Session(config=config)

or:

      gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=args.gpu_memory_fraction,allow_growth=True)
      sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options, log_device_placement=False))

来源：CSDN

作者：水野与小太郎

链接：https://blog.csdn.net/qq_36336522/article/details/86776236