解决问题：Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

最新推荐文章于 2022-03-29 16:58:35 发布

Kair94

最新推荐文章于 2022-03-29 16:58:35 发布

阅读量276

点赞数

CC 4.0 BY-SA版权

分类专栏： tensorflow

本文链接：https://blog.csdn.net/weixin_46474546/article/details/115316669

tensorflow 专栏收录该内容

7 篇文章

订阅专栏

博客提及让TensorFlow为显卡分配30%显存，还给出了‘Failed to get convolution algorithm. This is probably because cuDNN failed to initialize’错误的解决办法参考，以及Keras或TensorFlow显存占用问题的相关内容。

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

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