add pynvml for logging GPU metrics info in mlflow

falibabaei · web-flow · commit f1dd78ec1dc3 · 2025-02-03T09:47:35.000+01:00
diff --git a/requirements.txt b/requirements.txt
@@ -11,3 +11,4 @@ defusedxml~=0.5.0
 mlflow~=2.14.2
 GitPython~=3.1.43
 #gitpython
+pynvml==12.0.0
