Zainstaluj Tensorflow-GPU na WSL2
Czy ktoś pomyślnie zainstalował procesor graficzny Tensorflow-GPU na WSL2 z procesorami graficznymi NVIDIA? Mam Ubuntu 18.04 na WSL2, ale staram się zainstalować sterowniki NVIDIA. Każda pomoc byłaby doceniona, ponieważ jestem zagubiony.
Odpowiedzi
Mogę potwierdzić, że jestem w stanie uruchomić to bez potrzeby Dockera na WSL2 dzięki następującym artykułom:
https://qiita.com/Navier/items/cf551908bae707db4258
Pamiętaj, aby zaktualizować sterownik do wersji 460.15, która nie 455.41jest wymieniona w dokumentacji CUDA.
Uwaga, nie działa to z kartą w trybie TCC (tylko WDDM). Pamiętaj również, aby umieścić swoje pliki w systemie plików Linux (tj. Nie na dysku montowania, jak /mnt/c/). Wydajność jest znacznie szybsza w systemie plików Linux (ma to związek z różnicą w implementacji WSL 1 i WSL 2; patrz 1 , 2 i 3 ).
UWAGA: Zobacz także Czy wątek generatora klas (dziedziczenie sekwencji) jest bezpieczny w Keras / Tensorflow?