https://blog.csdn.net/xianqin_ma/article/details/79525519
具体的问题是下面所示:(我的TensorFlow版本是1.2)
E tensorflow/stream_executor/cuda/cuda_dnn.cc:359] could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
E tensorflow/stream_executor/cuda/cuda_dnn.cc:326] could not destroy cudnn handle: CUDNN_STATUS_BAD_PARAM
2018-03-12 10:55:56.078991: F tensorflow/core/kernels/conv_ops.cc:671] Check failed: stream->parent()->GetConvolveAlgorithms(&algorithms)
Aborted (core dumped)
对于此问题尝试了许多办法,最终的解决方案是:
执行 :sudo rm -R -f ~/.nv/ (一定最后边不要漏掉“/”,否则会提示“.nv”是目录)
其他方法我也有试过,但不行。
这个我也执行成功:
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.8)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
对于这个问题,我提供一下参考链接:https://github.com/tensorflow/tensorflow/issues/6698