diff --git a/batchflow/monitor.py b/batchflow/monitor.py
index 938accb20..974314341 100644
--- a/batchflow/monitor.py
+++ b/batchflow/monitor.py
@@ -318,7 +318,7 @@ class GPUResourceMonitor(ResourceMonitor):
     """ If the `CUDA_VISIBLE_DEVICES` is set, check it and return device numbers. Otherwise, return [0]. """
     def __init__(self, function=None, frequency=0.1, gpu_list=None, **kwargs):
         if pynvml is None:
-            raise ImportError('Install Python interface for pynvml')
+            raise ImportError('Install nvidia-ml-py to use GPU monitoring functionality')
         super().__init__(function=function, frequency=frequency, **kwargs)
 
         # Fallback to env variable
diff --git a/pyproject.toml b/pyproject.toml
index 107439e8d..0afcf2784 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -42,7 +42,7 @@ research = [
 
 monitor = [
     "psutil",
-    "pynvml"
+    "nvidia-ml-py"
 ]
 
 nn = [