[FIXED] Install Tensorflow-GPU on WSL2

Issue

Has anyone successfully installed Tensorflow-GPU on WSL2 with NVIDIA GPUs? I have Ubuntu 18.04 on WSL2, but am struggling to get NVIDIA drivers installed. Any help would be appreciated as I'm lost.

Solution

I can confirm I am able to get this working without the need for Docker on WSL2 thanks to the following article:

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

Be sure to update to driver version 460.15, not 455.41 as listed in the CUDA documentation.

Note, this does not work with the card in TCC mode (only WDDM). Also, be sure to place your files on the Linux file system (i.e. not on a mount drive, like /mnt/c/). Performance is significantly faster on the Linux file system (this has to do with the difference in implementation of WSL 1 vs. WSL 2; see 1, 2, and 3).

NOTE: See also Is the class generator (inheriting Sequence) thread safe in Keras/Tensorflow?

Answered By - A. Hendry