Installieren Sie die Tensorflow-GPU auf WSL2

Aug 31 2020

Hat jemand erfolgreich Tensorflow-GPU auf WSL2 mit NVIDIA-GPUs installiert? Ich habe Ubuntu 18.04 auf WSL2, habe aber Probleme, NVIDIA-Treiber zu installieren. Jede Hilfe wäre dankbar, da ich verloren bin.

Antworten

A.Hendry Sep 03 2020 at 00:26

Dank des folgenden Artikels kann ich bestätigen, dass ich dies ohne Docker auf WSL2 zum Laufen bringen kann:

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

Stellen Sie sicher, dass Sie auf die Treiberversion aktualisieren 460.15, die nicht 455.41in der CUDA-Dokumentation aufgeführt ist.

Beachten Sie, dass dies mit der Karte im TCC-Modus (nur WDDM) nicht funktioniert. Stellen Sie außerdem sicher, dass Sie Ihre Dateien auf dem Linux-Dateisystem ablegen (dh nicht auf einem Mount-Laufwerk, wie z /mnt/c/. B. ). Die Leistung des Linux-Dateisystems ist erheblich schneller (dies hängt mit dem Unterschied in der Implementierung von WSL 1 gegenüber WSL 2 zusammen; siehe 1 , 2 und 3 ).

HINWEIS: Siehe auch Ist der Thread des Klassengenerators (der die Sequenz erbt) in Keras / Tensorflow sicher?