Does `tensorflow-quantum` supports multiple GPU?
Shuhul24 opened this issue · comments
I am using TFQ for executing of quantum neural networks, but as I need it to be faster, I am trying to run it on multiple GPUs. Has tensorflow-quantum team did something regarding parallel computing for simulations of quantum circuits?
I have looked into this issue after googling and following are the issues links that I came up with:
https://stackoverflow.com/questions/73250190/does-tensorflow-quantum-support-gpu-and-if-so-how-do-i-make-it-use-mine
#599
#193
Some of the issues on these are open? Is there any update regarding this that we can compute parallely using tf.distribute.MirroredStrategy()?
Also adding the link of this issue (tensorflow/tensorflow#52988) as tensorflow==2.7.0 is one of the requisites for tensorflow-quantum.
I don't think GPU ended up being added at all (see the open PR: #687). That being said, if you want to accelerate your code across multiple nodes, you can do that. Here is a detailed tutorial on it: https://github.com/tensorflow/quantum/tree/research/qcnn_multiworker