这篇教程出现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 |