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

自学教程:出现Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR错误的解决办法dnn_rnn_ops.cc:1510 : Unknown: Fa

51自学网 2023-10-22 16:37:58
  TensorFlow
这篇教程出现Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR错误的解决办法dnn_rnn_ops.cc:1510 : Unknown: Fa写得很实用,希望能帮到您。
出现Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR错误的解决办法

问题分析:
出现Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR的原因,是电脑本身显存不足造成的,应该给予适当的空间。
解决办法:

# 引用tensorflow库
import tensorflow as tf

# 配置GPU参数
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))

 

如果上面方法解决不了,我用如下方法解决了:仅在需要时申请显存空间(程序初始运行时消耗很少的显存,随着程序的运行而动态申请显存);

import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices('GPU')
for gpu in gpus:
    tf.config.experimental.set_memory_growth(gpu, True)

返回列表
Anaconda的python虚拟环境中安装cudatoolkit和cudnn加速tensorflow
51自学网自学EXCEL、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。
京ICP备13026421号-1