Instal Tensorflow-GPU di WSL2

Aug 31 2020

Adakah yang berhasil menginstal Tensorflow-GPU di WSL2 dengan NVIDIA GPU? Saya memiliki Ubuntu 18.04 di WSL2, tetapi saya kesulitan untuk menginstal driver NVIDIA. Bantuan apa pun akan dihargai karena saya tersesat.

Jawaban

A.Hendry Sep 03 2020 at 00:26

Saya dapat mengonfirmasi bahwa saya bisa membuat ini berfungsi tanpa perlu Docker di WSL2 berkat artikel berikut:

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

Pastikan untuk memperbarui ke versi driver 460.15, tidak 455.41seperti yang tercantum dalam dokumentasi CUDA.

Catatan, ini tidak bekerja dengan kartu dalam mode TCC (hanya WDDM). Selain itu, pastikan untuk menempatkan file Anda pada sistem file Linux (bukan pada drive mount, seperti /mnt/c/). Kinerja secara signifikan lebih cepat pada sistem file Linux (ini berkaitan dengan perbedaan dalam implementasi WSL 1 vs WSL 2; lihat 1 , 2 , dan 3 ).

CATATAN: Lihat juga Apakah thread generator kelas (mewarisi Urutan) aman di Keras / Tensorflow?