Installa Tensorflow-GPU su WSL2

Question

Qualcuno ha installato con successo Tensorflow-GPU su WSL2 con GPU NVIDIA? Ho Ubuntu 18.04 su WSL2, ma sto lottando per installare i driver NVIDIA. Qualsiasi aiuto sarebbe apprezzato poiché mi sono perso.

Answer 1

Posso confermare di essere in grado di farlo funzionare senza la necessità di Docker su WSL2 grazie al seguente articolo:

https://qiita.com/Navier/items/cf551908bae707db4258

Assicurati di aggiornare alla versione del driver 460.15, non 455.41come elencato nella documentazione CUDA.

Nota, questo non funziona con la scheda in modalità TCC (solo WDDM). Inoltre, assicurati di posizionare i tuoi file sul file system Linux (cioè non su un'unità di montaggio, come /mnt/c/). Le prestazioni sono significativamente più veloci sul file system Linux (questo ha a che fare con la differenza nell'implementazione di WSL 1 rispetto a WSL 2; vedere 1 , 2 e 3 ).

NOTA: vedere anche Il generatore di classi (che eredita la sequenza) è sicuro per i thread in Keras / Tensorflow?