could not create cudnn handle: CUDNN

xiaoxiao2025-05-28  29

转载: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)

 

转载请注明原文地址: https://www.6miu.com/read-5030849.html

最新回复(0)