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

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

51自学网 2020-10-25 16:47:35

TensorFlow

这篇教程模型训练错误：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))

Tensorflow2.0学习（八） — tf.dataset自定义图像数据集
Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR 解决办法