解决 No se pudo crear el identificador cudnn: CUDNN_STATUS_INTERNAL_ERROR 问题

Motivo: memoria de video insuficiente (generalmente, si no está restringido, muchos códigos de marco de aprendizaje profundo se aplicarán a todo el espacio de memoria de video cuando se ejecute (incluso si no requiere tantos recursos, pero no permite otros programas después de solicitar it) Use), por lo que si ejecuta el código en este estado, habrá un problema de memoria de video insuficiente (porque hay otros programas u operaciones que requieren recursos de memoria de video))
Solución: agregue el siguiente código al script en ejecución

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

Supongo que te gusta

Origin blog.csdn.net/qq_28057379/article/details/115088287
Recomendado
Clasificación