Cuando se utiliza un archivo por lotes retrain.bat creación-V3-capacitación previa, encontramos el error "No se pudo obtener algoritmo de convolución. Esto es probablemente debido a que no se pudo inicializar cuDNN, a fin de tratar ..." retrain.py a continuación, abra el archivo, precedido por el siguiente código :
desde tensorflow.compat.v1 importación ConfigProto
de tensorflow.compat.v1 importación InteractiveSession
= ConfigProto config ()
config.gpu_options.allow_growth = verdad
la reunión = InteractiveSession (config = config)
el CD a archivos retrain.bat, ejecutar correctamente.
Artículo de referencia: https: //blog.csdn.net/qq_41868689/article/details/98503069
