Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.

Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
系统环境:Ubuntu16.04 ,TensorFlow1.13.1,cuda10.0, cudnnn7.6.5
问题描述:在运行pb文件进行测试的时候,出现这个问题,之前利用ckpt文件进行测试毫无问题;
原因分析:乍看时由于显存不够导致的,测试之后发现的却是(假像:是由于TensorFlow机制问题),但是我在更大的GPU机器上测试通过,反过来应该时TensorFlow贪心所致。
解决方法():
1、rm -rf ~/.nv(试过没用)
2、降低TensorFlow或者cudnn版本(别想了,会有其他各种问题,亲自做过,没啥用)
3、在session前面添加


```python
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.3
sess = tf.Session(config = config)
无用。已然灰心,试了多种,决定放弃的同时发现如果在
import TensorFlow as tf 之后加入以上代码:

```python
import tensorflow as tf
import numpy as np

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.3
sess = tf.Session(config = config)

成功运行,至于原因。。我也真不知道该说什么吗,估计有时候就是这么奇葩吧。

我叫谢琛,深度学习探索者一枚!(有意咨询xiec177)
tensorflow/pytorch交流群请加QQ:548368797
更多咨询咨询请关注!在这里插入图片描述

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

谢陈(xiec177)

致力于解决深度学习框架问题。

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值