Replace pynvml with nvidia-ml-py in dependencies and error messages

Co-authored-by: AlexeyKozhevin <[email protected]>

Author: Copilot

nbtools/core.py

@@ -155,7 +155,7 @@ def get_available_gpus(n=1, min_free_memory=0.9, max_processes=2, verbose=False,
     try:
         import pynvml
     except ImportError as exception:
-        raise ImportError('Install Python interface for nvidia_smi') from exception
+        raise ImportError('Install nvidia-ml-py package') from exception
 
     try:
         error_message = None
@@ -220,7 +220,7 @@ def get_gpu_free_memory(index, ratio=True):
     try:
         import pynvml
     except ImportError as exception:
-        raise ImportError('Install Python interface for nvidia_smi') from exception
+        raise ImportError('Install nvidia-ml-py package') from exception
 
     pynvml.nvmlInit()
     pynvml.nvmlDeviceGetCount()

setup.py

@@ -27,7 +27,7 @@
     zip_safe=False,
     platforms='any',
     install_requires=[
-        'pynvml>=11.5.0',
+        'nvidia-ml-py>=12.0.0',
         'blessed>=1.17',
         'psutil>=5.6',
         'requests>=2.24',