Instale Tensorflow-GPU em WSL2

Question

Alguém instalou com sucesso o Tensorflow-GPU no WSL2 com GPUs NVIDIA? Eu tenho Ubuntu 18.04 em WSL2, mas estou lutando para instalar os drivers NVIDIA. Qualquer ajuda seria apreciada, pois estou perdido.

Answer 1

Posso confirmar que consigo fazer isso funcionar sem a necessidade do Docker no WSL2, graças ao seguinte artigo:

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

Certifique-se de atualizar para a versão do driver 460.15, não 455.41conforme listado na documentação CUDA.

Observe que isso não funciona com a placa no modo TCC (apenas WDDM). Além disso, certifique-se de colocar seus arquivos no sistema de arquivos Linux (ou seja, não em uma unidade de montagem, como /mnt/c/). O desempenho é significativamente mais rápido no sistema de arquivos Linux (isso tem a ver com a diferença na implementação de WSL 1 vs. WSL 2; consulte 1 , 2 e 3 ).

NOTA: Consulte também O thread do gerador de classe (sequência herdada) é seguro no Keras / Tensorflow?