Utilisation des variables d'environnement

Slurm exporte un certain nombre de variables d'environnement relatives au Cluster, à la configuration générale et aux paramètres effectifs du Job. Ces variables sont accessibles dans les scripts et ont des noms très similaires aux options Slurm ou SBATCH correspondantes et sont donc facilement identifiables.

La liste complète des variables d’environnement est disponible à l’adresse : https://slurm.schedmd.com/sbatch.html#lbAG


Quelques variables utiles :

SLURM_JOB_PARTITION Partition utilisée (--partition)
SLURM_JOB_NAME Nom du Job (Attention, contrairement aux options output/error, les variables %j/%N... ne sont pas remplacées !)
SLURM_NTASKS Nombre de Tasks (--ntasks)
SLURM_CPUS_PER_TASK Nombre de CPUs par Task (--cpus-per-task)
SLURM_JOB_NUM_NODES Nombre de nœuds demandés/déduits (--nodes)
SLURM_JOB_NODELIST Liste des nœuds utilisés (--nodelist)

 


 

Pour récupérer une variable d'environnement 'VAR' en...

  • Shell/Bash/[T]CSH:  $VAR
  • Python: os.environ.get('VAR') [OU] os.getenv('VAR')
  • Java: System.getenv("VAR")
  • Lua: os.getenv(VAR)