I installed tensorflow-gpu 2.4.1 by the command ‘conda install tensorflow-gpu’, and it could recognize my gpu, that is, it will return True when I call tf.test.is_gpu_available().
But when I was installing tensorflow_federated by ‘pip install tensorflow_federated’, I found that my tensorflow-gpu installed by conda was uninstalled. And after the pip installation, tensorflow could not recognize my gpu anymore.
How can I use tensorflow_federated on GPU?