diff --git a/requirements.txt b/requirements.txt
index 535296e5..3fc07596 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -25,6 +25,7 @@ contextlib2==21.6.0
 jupyterlab>=3.6.2  # For transformer's visu
 IProgress>=0.4     # For jupyter with tdqm
 nested_lookup==0.2.25  # For lists management
+pynvml>=11.5.0
 
 # -------
 # Necessary but should be installed with scilpy (Last check: 01/2024):