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

could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

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

转载:https://blog.csdn.net/tangxianyu/article/details/80537942

could not create cudnn handle:CUDNN_STATUS_INTERNAL_ERROR

在CPU下运行没有问题,但是在GPU版本下,一直运行不成功。可以尝试限制一下GPU的运行资源,添加如下代码:

config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))
sess = tf.Session(config=config)

 

相关标签: gpu cudnn