1
두 개의 GPU가있는 노드에 --gres = gpu : 1 옵션을 사용하여 SLURM 작업을 제출하면 작업에 할당 된 GPU의 ID를 어떻게 얻을 수 있습니까? 이 목적을위한 환경 변수가 있습니까? 내가 사용하고있는 GPU는 모두 nvidia GPU입니다. 감사합니다. .여러 GPU 노드에서 SLURM 작업에 할당 된 GPU의 ID를 얻는 방법은 무엇입니까?
A
답변
2
환경 변수 CUDA_VISIBLE_DEVICES을 사용하여 GPU ID를 얻을 수 있습니다. 이 변수는 작업에 할당 된 GPU ID의 쉼표로 구분 된 목록입니다.
작동합니다. 감사. GPU_DEVICE_ORDINAL 환경 변수가 작동하는 것 같습니다. – Negelis