DeepSpeed

82ce4ae8 - Switch pynvml to nvidia-ml-py (#5529)

Switch pynvml to nvidia-ml-py (#5529) Fixes: #5517 Link to PyPI for nvidia-ml-py [here](https://pypi.org/project/nvidia-ml-py/) showing usage remaining the same as previous pynvml package.