Desconhecido: Falha ao obter o algoritmo de convolução. Provavelmente porque cuDNN falhou ao inicializar ...

Encontrei o problema acima porque o código que estava executando exigia tensorflow 1.9.0, mas o meu era 1.14.0. Verifiquei muitos métodos para fazer o downgrade da versão, então adotei mais tarde o seguinte método e coloquei-o no topo do código para resolvê-lo:

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

 

Acho que você gosta

Origin blog.csdn.net/weixin_45191152/article/details/103715785
Recomendado
Clasificación