pytorch
e7e51b3a - Fix NVML visible device parsing (#92315)

Commit
2 years ago
Fix NVML visible device parsing (#92315) `CUDA_VISIBLE_DEVICES` can contain either ordinals or UUIDs Extend the logic to be able to parse it by UUID Added unit test to validate that parser and matcher behavior matches that of 525.60.13 driver Skip MIG- device parsing Fixes https://github.com/pytorch/pytorch/issues/90543 Pull Request resolved: https://github.com/pytorch/pytorch/pull/92315 Approved by: https://github.com/ngimel
Author
Committer
Parents
Loading