您的位置:首页 > 其它

could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR问题解决方案

2018-03-12 13:37 489 查看
在准备用SRGAN对数据集进行超分实验时,一路都是坑呀!!!!!!!!!!!
下面我会就我遇到的问题,提供解决方案(以下方案是我应用之后,成功才给予分享的):
1. AttributeError: ' module' object has no attribute 'imread':



这是在读取图片时遇到的问题,对于此问题,我的解决办法是安装pillow包。
具体方案:  pip install  pillow
执行完成之后,我的可以成功执行原先代码继续。
2. could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

具体的问题是下面所示:(我的TensorFlow版本是1.2)E tensorflow/stream_executor/cuda/cuda_dnn.cc:359] could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERRORE 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 -f ~/.nv/    (一定最后边不要漏掉“/”,否则会提示“.nv”是目录)
其他方法我也有试过,但不行。
其他人的方案(我并执行没有成功):
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.1)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

对于这个问题,我提供一下参考链接:https://github.com/tensorflow/tensorflow/issues/6698
欢迎大家留下相关问题,大家一起讨论。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐