TensorFlow Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
Se stai usando Tensorflow con una gpu ( nel mio caso nvdia 2080Ti) probabilmente ti sei scontrato con l’errore “Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.” nel momento in cui stai provando a fare il fitting della rete. Il problema nasce da un’incompatibilità di versioni in cuda e cudnn e nei driver nvidia. Sul tema ho fatto un pò di test ed ho verificato che risolvendo i problemi di compatibilità di versione, con tensorflow 2.3.0-gpu non ci ho avuto problemi di sorta, mentre ho riscontrato che il problema persisteva nella versione tensorflow2.2.0-gpu.
Per risolvere il problema è stato sufficiente aggiungere nel codice, dopo avere importato tensorflow, le seguenti righe di configurazione, valide rispettivamente per tensorflow V2 e tensorflow V1.
TensorFlow V2
physical_devices = tf.config.experimental.list_physical_devices(‘GPU’)
tf.config.experimental.set_memory_growth(physical_devices[0], True)
TensorFlow V1
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)
Sono amante della tecnologia e delle tante sfumature del mondo IT, ho partecipato, sin dai primi anni di università ad importanti progetti in ambito Internet proseguendo, negli anni, allo startup, sviluppo e direzione di diverse aziende; Nei primi anni di carriera ho lavorato come consulente nel mondo dell’IT italiano, partecipando attivamente a progetti nazionali ed internazionali per realtà quali Ericsson, Telecom, Tin.it, Accenture, Tiscali, CNR. Dal 2010 mi occupo di startup mediante una delle mie società techintouch S.r.l che grazie alla collaborazione con la Digital Magics SpA, di cui sono Partner la Campania, mi occupo di supportare ed accelerare aziende del territorio .
Attualmente ricopro le cariche di :
– CTO MareGroup
– CTO Innoida
– Co-CEO in Techintouch s.r.l.
– Board member in StepFund GP SA
Manager ed imprenditore dal 2000 sono stato,
CEO e founder di Eclettica S.r.l. , Società specializzata in sviluppo software e System Integration
Partner per la Campania di Digital Magics S.p.A.
CTO e co-founder di Nexsoft S.p.A, società specializzata nella Consulenza di Servizi in ambito Informatico e sviluppo di soluzioni di System Integration, CTO della ITsys S.r.l. Società specializzata nella gestione di sistemi IT per la quale ho partecipato attivamente alla fase di startup.
Sognatore da sempre, curioso di novità ed alla ricerca di “nuovi mondi da esplorare“.
Comments