当前位置: 代码迷 >> 综合 >> tensorflow卷积报错Failed to get convolution algorithm. This is probably because cuDNN failed t
  详细解决方案

tensorflow卷积报错Failed to get convolution algorithm. This is probably because cuDNN failed t

热度:17   发布时间:2023-12-21 02:36:22.0

tensorflow卷积报错Failed to get convolution algorithm. This is probably because cuDNN failed t

说明:环境是tensorflow,测试GPU没有问题。

在tensorflow2.0和tensorflow1.15我都有遇到这个错误,主要是显卡内存分配问题。

但在运行模型出现错误。

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]

解决方法添加下面两行

import osos.environ['CUDA_VISIBLE_DEVICES'] = '/gpu:0'  # 运行程序,都会占用gpu0全部资源# 多个GPU时,如果运行只使用了一个的话,可以设置为‘/gpu:0,1’等等

# 另一种写法
os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID' # 按照PCI_BUS_ID顺序从0开始排列GPU设备
os.environ['CUDA_VISIBLE_DEVICES'] = "0,1" #设置当前使用的GPU设备为0,1号两个设备,名称依次为'/gpu:0'、'/gpu:1'。
#[0,1]和[1,0]排列的设备是不同的,排在前面的设备优先级高,运行程序的时候会优先使用。

其他一些解答(我测试没有成功,但可能是我遇到的情况不一致)

1.矛头指向了是因为显存分配没搞好造成的

解决代码

import tensorflow as tf
config = tf.compat.v1.ConfigProto(allow_soft_placement=True)
config.gpu_options.per_process_gpu_memory_fraction = 0.8
tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))

来分析一下这段代码以及问题原因:
众所周知,keras以TensorFlow做后台时,TensorFlow会默认吃掉所有可用的显存(即便此时没有用到做计算,这些显存也会显示占用来待命)

第二行的0.3,意思是限制每个模型只能分配30%的显存,不允许吃掉所有。当然这是理论值,当实际开始计算时,如果需要,依然可以超出这个理论值。

2 也是说显卡内存分配问题

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

这个设置为按需分配

  相关解决方案