diff --git a/requirements.txt b/requirements.txt
index 73d86f1..6a4ba83 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,4 +4,4 @@ torch
 transformers
 datasets
 peft
-pynvml
\ No newline at end of file
+pynvml>=11.5.0
\ No newline at end of file