pytorch
81da50a9 - Return device count using nvml (#84879)

Commit
2 years ago
Return device count using nvml (#84879) Fixes https://github.com/pytorch/pytorch/issues/83973 Pull Request resolved: https://github.com/pytorch/pytorch/pull/84879 Approved by: https://github.com/ngimel
Author
Committer
Parents
Loading