欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

程序员文章站 2022-05-26 23:44:09
...

经过观察我的原因是,显存不足

使用的方法是,限制最大显存使用比率

在程序中加入如下几行进行限制:

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.75 #比值可以根据具体情况调节
session = tf.Session(config=config)

然后问题就解决了