hello guys! thank your reading this.does anyone know how to run pennylane code on torch.cuda , thank you if know this,much appreciate!

Hi @oceaneyyys The issue here seems to be related to your torch input data. When casting the labels in your loop from an integer array to a float array, the data then resides on the CPU. I made some modifications to the locality of your data in the provided script, and this allows it to run end to…

Pennylane running on torch.cuda?

PennyLane Help

oceaneyyys April 5, 2024, 5:28am 8

thank you for your reply!

Topic		Replies	Views
PennyLane v0.19.0 + CUDA PennyLane Help	7	861	November 26, 2021
RuntimeError with Mixed Device Tensors when Integrating PennyLane with PyTorch PennyLane Help	3	752	March 26, 2024
Problems with using lightning.gpu PennyLane Help	6	352	January 12, 2024
How to get the matrix of circuit in cuda? PennyLane Help	8	660	November 15, 2023
Transfer learning error PennyLane Help	42	3726	January 26, 2024

Pennylane running on torch.cuda?

Related topics