Instalar Tensorflow-GPU en WSL2

Question

¿Alguien ha instalado con éxito Tensorflow-GPU en WSL2 con GPU NVIDIA? Tengo Ubuntu 18.04 en WSL2, pero estoy luchando para instalar los controladores NVIDIA. Cualquier ayuda sería apreciada porque estoy perdido.

Answer 1

Puedo confirmar que puedo hacer que esto funcione sin la necesidad de Docker en WSL2 gracias al siguiente artículo:

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

Asegúrese de actualizar a la versión del controlador 460.15, no a la 455.41que figura en la documentación de CUDA.

Tenga en cuenta que esto no funciona con la tarjeta en modo TCC (solo WDDM). Además, asegúrese de colocar sus archivos en el sistema de archivos de Linux (es decir, no en una unidad de montaje, como /mnt/c/). El rendimiento es significativamente más rápido en el sistema de archivos de Linux (esto tiene que ver con la diferencia en la implementación de WSL 1 frente a WSL 2; consulte 1 , 2 y 3 ).

NOTA: Consulte también ¿Es seguro el subproceso del generador de clases (que hereda la secuencia) en Keras / Tensorflow?