[UPDATE] requirements with pynvml

Author: dawmro

requirements.txt

@@ -95,6 +95,7 @@ ninja==1.11.1.4
 notebook==7.2.2
 notebook_shim==0.2.4
 numpy==1.26.4
+nvidia-ml-py==12.575.51
 opt_einsum==3.4.0
 optree==0.16.0
 overrides==7.7.0
@@ -114,6 +115,7 @@ pyarrow==19.0.1
 pycparser==2.22
 Pygments==2.18.0
 pylzma==0.5.0
+pynvml==12.0.0
 pyparsing==3.1.4
 python-dateutil==2.9.0.post0
 python-json-logger==2.0.7