Is this different than the existing multi_gpu_model? (https://keras.io/utils/#multi_gpu_model)
Somewhat off topic: I'm rediscovering the dance required to set up a new Windows 10 laptop with Python, CUDA, and TensorFlow. If I understand correctly, pyenv does not work on Windows. Is there something similar?
Also, if I installed TensorFlow in WSL, it wouldn't be able to use the GPU, would it?
In my experience conda is the way to go for setting up your environment. I did not have great luck with pyenv, YMMV.
I don't know about WSL. Maybe setup a dual boot with Ubuntu on your system?