【解决方案】tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm

背景

tensorflow.python.framework.errors_impl.UnknownError: 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. [Op:Conv2D]
在这里插入图片描述
更换了tensorflow版本,在使用tensorflow-gpu 2.0.1时候,遇到的这个问题。
环境:
tensorflow-gpu 2.0.1 (安装了anaconda,使用的pip安装,清华镜像)
Windows 10
CUDA 10.0
cuDNN 7.6.4

解决方法

有的人认为,利用pip安装的包,会直接帮你安装好相关依赖包,例如我在本地上事先安装好cudnn和cuda,但pip install tensorflow-gpu的时候会再次帮你装一个cuda 和 cudnn,这个时候cuda和cudnn的版本会比较低,造成与tf-gpu版本不兼容,他们试着把conda种的cuda和cudnn卸载掉,然后确实也成功了。但是我在conda list看cuda和cudnn版本的时候,他们与tf-gpu的版本实际上是兼容的。于是找到了下面这个方法。似乎与显卡内存分配和Session交互方式有关联??

在代码前面加一段

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

2020.4.9更新一下

今天又遇到了这个问题,是在跑Yolov3-keras的时候,我使用了1个分类的行人数据,里面又500+张标注好的行人数据,然后epoch有500,报错了显示说run out of memory,这个时候我把batch_size从16调为了4,还是同样的错误,又调整为1,这时候就不报错了。

参考:https://github.com/tensorflow/tensorflow/issues/24828

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章