gpt4 book ai didi

slurm - 如何在程序中获取 SLURM 任务 ID

转载 作者:行者123 更新时间:2023-12-02 19:49:01 24 4
gpt4 key购买 nike

我正在运行 srun -n 100 python foo.py。在 python 脚本中,它如何找出它是哪个任务编号/id/rank?是否设置了环境变量?

最佳答案

查看 man srunman sbatch 以获得环境变量列表。 $SLURM_PROCID 可能是您需要的。

关于slurm - 如何在程序中获取 SLURM 任务 ID,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/58648721/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com