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

could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

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

could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

    把gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
    sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
    
    改成gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.3)
    sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
   或者再小一点