Migrate the pynvml package to nvidia-ml-py

Address warning:
```
core/gpu.py:26: FutureWarning: The pynvml package is deprecated. Please install nvidia-ml-py instead. If you did not install pynvml directly, please report this to the maintainers of the package that installed pynvml for you
```