gpt4 book ai didi

gpu - 如何在多 GPU 节点上获取分配给 SLURM 作业的 GPU ID?

转载 作者:行者123 更新时间:2023-12-01 19:41:46 40 4
gpt4 key购买 nike

当我使用选项 --gres=gpu:1 向具有两个 GPU 的节点提交 SLURM 作业时,如何获取为该作业分配的 GPU ID?是否有用于此目的的环境变量?我使用的 GPU 都是 nvidia GPU。谢谢。

最佳答案

您可以通过环境变量CUDA_VISIBLE_DEVICES获取GPU ID。该变量是分配给作业的 GPU ID 的逗号分隔列表。

关于gpu - 如何在多 GPU 节点上获取分配给 SLURM 作业的 GPU ID?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43967405/

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