排坑：tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This

最新推荐文章于 2023-03-27 12:57:27 发布

TFATS

最新推荐文章于 2023-03-27 12:57:27 发布

阅读量3.5k

点赞数 5

分类专栏： tensorflow keras 排坑文章标签： tensorflow-gpu UnknownError cuDNN初始化 GPU内存分配 allow_growth

本文链接：https://blog.csdn.net/TFATS/article/details/113978075

版权

keras 同时被 3 个专栏收录

38 篇文章 2 订阅

订阅专栏

tensorflow

35 篇文章 11 订阅

订阅专栏

排坑

4 篇文章 0 订阅

订阅专栏

一，问题描述

在使用tensorflow-gpu时，出现下面的错误：

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.
[[node sequential/conv2d/Conv2D (defined at tf_keras_classification_model-cnn.py:107) ]] [Op:__inference_distributed_function_930]
Function call stack:
distributed_function

二，解决方法

使用tensorflow，如果不加设置，即使是很小的模型也会占用整块GPU，造成资源浪费。所以我们需要设置，使程序按需使用GPU。

在代码前添加如下几行代码，实现tensorflow按需分配GPU

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
# 定义TensorFlow配置
config = ConfigProto()
# 配置GPU内存分配方式，按需增长，很关键
config.gpu_options.allow_growth = True
# 在创建session的时候把config作为参数传进去
session = InteractiveSession(config=config)