Installez Tensorflow-GPU sur WSL2

Aug 31 2020

Quelqu'un a-t-il installé avec succès Tensorflow-GPU sur WSL2 avec des GPU NVIDIA? J'ai Ubuntu 18.04 sur WSL2, mais j'ai du mal à installer les pilotes NVIDIA. Toute aide serait appréciée car je suis perdu.

Réponses

A.Hendry Sep 03 2020 at 00:26

Je peux confirmer que je peux faire fonctionner cela sans avoir besoin de Docker sur WSL2 grâce à l'article suivant:

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

Assurez-vous de mettre à jour la version du pilote 460.15, pas 455.41comme indiqué dans la documentation CUDA.

Notez que cela ne fonctionne pas avec la carte en mode TCC (uniquement WDDM). Assurez-vous également de placer vos fichiers sur le système de fichiers Linux (c'est-à-dire pas sur un lecteur de montage, comme /mnt/c/). Les performances sont nettement plus rapides sur le système de fichiers Linux (cela a à voir avec la différence d'implémentation de WSL 1 par rapport à WSL 2; voir 1 , 2 et 3 ).

REMARQUE: Voir également Le générateur de classe (héritant de la séquence) est-il thread-safe dans Keras / Tensorflow?