问题:
2019-05-28 10:19:53.732267: E tensorflow/stream_executor/cuda/cuda_dnn.cc:385]
2019-05-28 10:19:53.732353: E tensorflow/stream_executor/cuda/cuda_dnn.cc:352] could not destroy cudnn handle: CUDNN_STATUS_BAD_PARAM
2019-05-28 10:19:53.732374: F tensorflow/core/kernels/conv_ops.cc:717] Check failed: stream->parent()->GetConvolveAlgorithms( conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms)
解决方案:
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction =0.7 ----->减小这个比例值
set_session(tf.Session(config=config))