pytorch

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

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