Libraries for Pennylane GPU

Do I need to download any special libraries for my quantum circuit to work on the GPU? I am using Python 3.8 and using Nvidia DGX-1.

Hi @mass_of_15
If you are on a DGX-1 (V100 or newer), and your CUDA libraries are on your path, then you should be able to set-up a working env with

python -m venv myenv && source ./myenv/bin/activate
python -m pip install pennylane pennylane-lightning[gpu] cuquantum

All instructions for using lightning.gpu as the device should work natively. Hope this helps!