search

tensorflow / quantum

Hybrid Quantum-Classical Machine Learning in TensorFlow
https://www.tensorflow.org/quantum
Apache License 2.0
1.77k stars 558 forks source link

GPU support in TFQ #790

Open masonca000 opened 1 year ago

masonca000 commented 1 year ago

Hi,

I am trying to parallelize a hybrid network using a gpu, but I'm unsure if TFQ supports gpus yet. I have seen previous issues that said TFQ does not support gpus but that it was being looked into. Is there any update on that?

lockwo commented 1 year ago

GPU support is in progress. You might be able to install it from https://github.com/tensorflow/quantum/pull/759 (although there are some issues I have heard of), but you can see a number of PRs not yet closed in relation to this https://github.com/tensorflow/quantum/pulls.