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

在准备用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_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 -rf ~/.nv/    (一定最后边不要漏掉“/”,否则会提示“.nv”是目录, 另外“~”前边有空格)

这里解释一下:~/.nv/文件夹下是一些缓存信息,所以这一步的目的就是删除~/.nv/内的缓存信息。

那么释放缓存的方式还有一种方式就是重启电脑(感谢@XSD995366159的反馈)

所以执行sudo rm -rf ~/.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

欢迎大家留下相关问题,大家一起讨论。

 

 

  • 19
    点赞
  • 40
    收藏
    觉得还不错? 一键收藏
  • 63
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 63
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值