tensorflow 1.4 cuda8.0 cudnn 6.0
执行gpu版本的 MNIst 时候报错:
E tensorflow/stream_executor/cuda/cuda_dnn.cc:371] could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
E tensorflow/stream_executor/cuda/cuda_dnn.cc:338] could not destroy cudnn handle: CUDNN_STATUS_BAD_PARAM
F
tensorflow/core/kernels/conv_ops.cc:672] Check failed:
stream->parent()->GetConvolveAlgorithms(
conv_parameters.ShouldIncludeWinogradNonfusedAlgo(),
&algorithms)
试了很多办法
(1):
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))这个方法对我来说不管用,还是之前的错误;
(2):执行sudo rm -f ~/.nv 解决问题
参考:https://github.com/tensorflow/tensorflow/issues/6698