Back to Devices

lightning.gpu

  • Built by PennyLane
  • QJIT
  • Large Workloads
  • Distributed Simulation
  • Performance
  • GPU (simulator)
  • Linux

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:

See all Lightning devices:

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 qml
dev = qml.device('lightning.gpu', wires=20) 

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