Pennylane

Pennylane on avoimen lähdekoodin monialustainen Python-kirjasto kvanttikoneoppimiseen, kvanttikemiaan ja kvanttilaskentaan. Pennylane-lightning on suorituskykyinen tilavektorisimulaattori Pennylanelle. Se tukee tällä hetkellä GPU:ita LUMIssa Kokkos -kehyksen kautta.

Saatavilla

Tällä hetkellä tuetut Pennylane-versiot:

Version	Module	LUMI	Notes
0.43.0	`pennylane-lightning/0.43.0-gpu`	X	default version
0.42.0	`pennylane-lightning/0.42.0-gpu`	X
0.41.1	`pennylane-lightning/0.41.1-gpu`	X

Kaikki moduulit perustuvat Tykkyyn ja käyttävät LUMI-container-wrapperia. Tarjolla on wrapper-skriptejä, jotta yleiset komennot kuten python, python3, pip ja pip3 toimivat normaalisti. Lisätietoja on sivulla LUMI container wrapper.

Lisenssi

Pennylane on lisensoitu Apache License 2.0 -lisenssillä.

Käyttö

Jos haluat käyttää Pennylanen oletusversiota LUMIssa, alusta se komennolla:

module use /appl/local/quantum/modulefiles

ja

module load pennylane-lightning

Jos haluat käyttää tiettyä versiota (katso yllä saatavilla olevat versiot), käytä komentoa:

module load pennylane-lightning/0.43.0-gpu

missä 0.43.0-gpu on määritetty versio

Tämä komento näyttää myös kaikki saatavilla olevat versiot:

module avail pennylane-lightning

Esimerkkiajoskripti

Esimerkkiajoskripti yhden GPU:n ja yhden CPU-ytimen varaamiseen yhdellä solmulla:

LUMI

#!/bin/bash
#SBATCH --account=<project>
#SBATCH --partition=dev-g
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --mem=128G
#SBATCH --time=0:10:00
#SBATCH --gpus=1
#SBATCH --job-name=pennylane-example

# setting environment variables to specify how the OpenMP threads in a program are bound to processors
export OMP_PROC_BIND=spread  
export OMP_PLACES=threads

module use /appl/local/quantum/modulefiles # or module load Local-quantum
module load pennylane-lightning
python <file_name>.py

Lähetä skripti komennolla sbatch <script_name>.sh