diff --git a/requirements.txt b/requirements.txt
index f49ec035b..79905fa7d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -19,4 +19,4 @@ evaluate==0.4.0
 rouge-score==0.1.2
 scipy
 scikit-learn==1.2.2
-nvidia-ml-py3
+pynvml