GPU map needs all 8 leading 0s..

This commit is contained in:
Joachim Meyer 2023-02-08 09:10:25 +01:00
parent 063713d18c
commit 4259611495

View File

@ -28,6 +28,6 @@ if __name__ == "__main__":
gpus = slot["AssignedGPUs"].split(',') gpus = slot["AssignedGPUs"].split(',')
for gpu_id in gpus: for gpu_id in gpus:
gpu = slot["GPUs_" + gpu_id.strip().replace("-", "_")] gpu = slot["GPUs_" + gpu_id.strip().replace("-", "_")]
gpu_map[gpu["Id"]] = gpu["DevicePciBusId"] gpu_map[gpu["Id"]] = "0000" + gpu["DevicePciBusId"]
slot_gpu_map[machine] = gpu_map slot_gpu_map[machine] = gpu_map
print(json.dumps(slot_gpu_map)) print(json.dumps(slot_gpu_map))