lightning.gpu

lightning.gpu is PennyLane’s GPU-extension of lightning.qubit; it’s our most performant all-around device for heavy workloads, made possible by NVIDIA’s cuQuantum SDK.

Recommended for:

• Large state vector simulations with 20+ qubits.
• Workflows that may benefit from parallelization over multiple GPUs.
• Fast gradients with parallel adjoint differentiation.
• Linux operating systems.

Documentation

To learn more, please visit the device documentation:

• lightning.gpu documentation

See all Lightning devices:

• lightning.qubit
• lightning.kokkos
• lightning.amdgpu
• lightning.gpu
• lightning.tensor

Installation

The lightning.gpu device can be installed with:

pip install pennylane-lightning-gpu

For more details on installation and dependencies, visit the Install PennyLane page.

Device Initialization

Initialize the device in PennyLane with:

import pennylane as qp
dev = qp.device('lightning.gpu', wires=20) 

For more details on device settings and keyword arguments, see the device documentation.