slurm comment obtenir le cpu_ids d'un travail de sbatch?

Question

Quelle est la meilleure façon d'obtenir les ID de cpu valides dans un travail en cours? Mon idée est de faire une allocation -> enveloppe une commande docker avec les limites de l'allocation -> exécuter nvidia-docker sur un serveur gpu distant.

Pour limiter le docker à l'allocation, je dois lui indiquer le cpu_ids.

Ma soumission d'emploi ressemblera:

sbatch -o test.txt -c2 -n 10 --mem=10GB --wrap="job that needs the cpu_ids" 

Answer 1

Dans le script que vous lancez avec sbatch, vous devez exécuter la ligne suivante afin d'obtenir les ID de CPU et de les enregistrer dans un fichier:

scontrol -dd show job $SLURM_JOB_ID | grep "CPU_IDs" | sed 's/^ *//' > cpuids.out