Desconocido: no se pudo obtener el algoritmo de convolución. Esto probablemente se deba a que cuDNN no se pudo inicializar ...

Encontré el problema anterior porque el código que estaba ejecutando requería tensorflow 1.9.0, pero el mío era 1.14.0. Verifiqué muchos métodos para degradar la versión, así que luego adopté el siguiente método y lo puse en la parte superior del código para resolverlo:

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/weixin_45191152/article/details/103715785
Recomendado
Clasificación