slurm如何從一個sbatch作業中獲取cpu_ids？

Question

從正在運行的作業中獲取有效CPU ID的最佳方式是什麼？ 我的想法是做一個分配 - >用分配限制包裝docker命令 - >在遠程gpu服務器上運行nvidia-docker。

要限制碼頭到分配我需要告訴它cpu_ids。

我的作業提交的樣子：

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

Answer 1

在你sbatch啓動腳本，您應該按順序執行的下一行，以獲得CPU的ID，並將它們保存在一個文件中：

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