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
更多咨询咨询请关注!