出现Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
错误的解决办法
问题分析:
出现Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
的原因,是电脑本身显存不足造成的,应该给予适当的空间。
解决办法:
- 配置GPU参数
(借鉴了其他博主的答案,作为首要参考)
在代码的开头部分,直接添加以下代码,用于配置GPU使用信息:
import tensorflow as tf
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))
- 针对于初学者
在开始训练模型的代码中,注意batch_size
的大小,我的GTX1650的本本只能卑微到用batch_size=1
。
model.fit(x_train, y_train, batch_size=1, epochs=15)
(开始跑项目的时候,是直接把别人的代码弄来跑,环境都配好了,就是报错,仔细捋了一下过程,发现是batch_size=32
造成的,改为batch_size=1
就可以正常训练了)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)