问题
2018-04-27 17:39:18.270790: E T:\src\github\tensorflow\tensorflow\stream_executor\cuda\cuda_dnn.cc:403] could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
2018-04-27 17:39:18.270961: E T:\src\github\tensorflow\tensorflow\stream_executor\cuda\cuda_dnn.cc:370] could not destroy cudnn handle: CUDNN_STATUS_BAD_PARAM
2018-04-27 17:39:18.271076: F T:\src\github\tensorflow\tensorflow\core\kernels\conv_ops.cc:712] Check failed: stream->parent()->GetConvolveAlgorithms( conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms)
1
2
3
解决:
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.1)
with tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)) as sess:
本文详细记录了一次使用TensorFlow时遇到的CUDA DNN错误,具体表现为无法创建和销毁cuDNN handle的问题,并附带了内部错误和参数错误的详细日志。通过调整GPU内存分配比例,成功解决了这一问题。
1万+

被折叠的 条评论
为什么被折叠?



