one solution is `pip install tensorflow[and-cuda]` in the env you want to use the gpu or... `pip install -U tensorflow[and-cuda]`