使用gpu运算卷积网络时报错Failed to get convolution algorithm. This is probably because cuDNN....
发布于2019-08-17 20:32 阅读(2495) 评论(0) 点赞(1) 收藏(0)
在我们完成tensorflow-gpu配置后，我们希望使用其做一些实践，例如我希望可以使用它来加快之前跑起来极慢的facenet人脸识别实践。
配置参数：
cpu：E5-2697 v3
gpu：RTX 2080 TI
内存：64G
数据量：55万张
python：3.7
tensorflow：1.13.1
当我使用--gpu_memory_fraction 0.9控制gpu内存使用率为90%时报错Failed to get convolution algorithm. This is probably because cuDNN failed to initialize.
经过不断地踩坑总结以下几种方法解决这一问题：
(1) 降低tensorflow-gpu版本(不推荐，会导致后期一系列问题，解决起来比较费心神)
(2) 针对ipython可是使用restart kernel实现(不推荐，因为不一定成功)
(3) 减小gpu内存使用率，例如将命令改为--gpu_memory_fraction 0.5(亲测有效，但可能只是针对我的情况)
所属网站分类: 技术文章 > 博客
作者：美美哒
链接：https://www.pythonheidong.com/blog/article/48424/03ab42d3ea6d112a7d4e/
来源：python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容：(最多支持255个字符)
---无人问津也好，技不如人也罢，你都要试着安静下来，去做自己该做的事，而不是让内心的烦躁、焦虑，坏掉你本来就不多的热情和定力