ubantu16.04 tensorflow2.0 cuda10.1 环境下出现：Failed to get convolution algorithm.

luoziyuan1111

于 2020-04-09 17:24:45 发布

阅读量241

点赞数

CC 4.0 BY-SA版权

文章标签： tensorflow

本文链接：https://blog.youkuaiyun.com/luoziyuan1111/article/details/105415472

本文详细介绍了在Ubuntu16.04系统下使用TensorFlow2.0和CUDA10.1时遇到的显存溢出问题，并提供了有效的解决方案。通过调整GPU显存增长设置，可以避免Failed to get convolution algorithm错误，确保深度学习任务的顺利运行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

ubantu16.04 tensorflow2.0 cuda10.1 环境下出现：Failed to get convolution algorithm.

经过排查发现是因为可能是显卡内存溢出导致的错误在代码中添加如下代码即可避免：

physical_devices = tf.config.experimental.list_physical_devices('GPU')
assert len(physical_devices) > 0, "Not enough GPU hardware devices available"
tf.config.experimental.set_memory_growth(physical_devices[0], True)

http://www.bubuko.com/infodetail-3235806.html
https://blog.youkuaiyun.com/qq_41868689/article/details/98503069