Wenn eine Batch - Datei mit retrain.bat Anfang-V3-Training vor, fanden wir den Fehler „Fehler Faltungsalgorithmus zu erhalten. Dies ist wahrscheinlich , weil cuDNN konnte nicht initialisiert werden , so versuchen ...“ retrain.py öffnen Sie die Datei mit dem folgenden Code voran :
von tensorflow.compat.v1 Import ConfigProto
von tensorflow.compat.v1 Import InteractiveSession
= ConfigProto config ()
config.gpu_options.allow_growth = True
die session = InteractiveSession (config = config)
die CD zu retrain.bat Datei erfolgreich ausgeführt.
Referenzartikel: https: //blog.csdn.net/qq_41868689/article/details/98503069
