您当前的位置：首页 > IT编程 > 深度学习
\| C语言 \| Java \| VB \| VC \| python \| Android \| TensorFlow \| C++ \| oracle \| 学术与代码 \| cnn卷积神经网络 \| gnn \| 图像修复 \| Keras \| 数据集 \| Neo4j \| 自然语言处理 \| 深度学习 \| 医学CAD \| 医学影像 \| 超参数 \| pointnet \| pytorch \| 异常检测 \| Transformers \| 情感分类 \| 知识图谱 \|

自学教程：模型训练错误：Failed to get convolution algorithm，This is probably because cuDNN failed to initialize.

51自学网 2020-11-03 16:27:51

深度学习

这篇教程模型训练错误：Failed to get convolution algorithm，This is probably because cuDNN failed to initialize.写得很实用，希望能帮到您。

解决方法：

在训练的脚本开头添加以下代码进行动态分配内存，然后重新运行就可以。

import tensorflow as tf
import keras
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
keras.backend.tensorflow_backend.set_session(tf.Session(config=config))

numpy将数组保存为文件
Keras: Multiple outputs and multiple losses