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)

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

你可能感兴趣的:(Tensorflow)