Установите Tensorflow-GPU на WSL2

Question

Кто-нибудь успешно установил Tensorflow-GPU на WSL2 с графическими процессорами NVIDIA? У меня Ubuntu 18.04 на WSL2, но мне не удается установить драйверы NVIDIA. Любая помощь будет оценена, так как я потерялся.

Answer 1

Я могу подтвердить, что могу заставить это работать без необходимости в Docker на WSL2, благодаря следующей статье:

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

Обязательно обновите драйвер до версии 460.15, не 455.41указанной в документации CUDA.

Обратите внимание, это не работает с картой в режиме TCC (только WDDM). Кроме того, не забудьте разместить файлы в файловой системе Linux (т.е. не на монтируемом диске, например /mnt/c/). Производительность файловой системы Linux значительно выше (это связано с различием в реализации WSL 1 и WSL 2; см. 1 , 2 и 3 ).

ПРИМЕЧАНИЕ. См. Также. Является ли генератор классов (наследующий последовательность) потокобезопасным в Keras / Tensorflow?