diff --git a/requirements.txt b/requirements.txt
index 7a4638f..86c86b0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1 +1 @@
-pynvml==11.5.0
+nvidia-ml-py==12.560.30