pynvml deprecation

Due to pynvml (https://pypi.org/project/pynvml/) being deprecated, this error occurs from PyTorch:
```
site-packages\torch\cuda\__init__.py:65: 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.
  import pynvml  # type: ignore[import]
```