如何从一个sbatch作业中获取cpu_ids？

Question

从正在运行的作业中获取有效cpu id的最佳方法是什么？ 我的想法是做一个分配 - ＆gt;用分配限制包装一个docker命令 - ＆gt;在远程gpu服务器上运行nvidia-docker。

要将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