当前位置: 代码迷 >> 综合 >> 【解决】“Failed to get convolution algorithm. This is probably because cuDNN failed to initialize”
  详细解决方案

【解决】“Failed to get convolution algorithm. This is probably because cuDNN failed to initialize”

热度:90   发布时间:2023-09-22 07:10:16.0

问题描述:

“Failed to get convolution algorithm. This is probably because cuDNN failed to initialize”

解决方案:

不加这几句,则CONV 报错

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)
  相关解决方案