feat: use nvidia-ml-py and freeze version

hhzhang16 · hhzhang16 · commit 43fabbb40274 · 2025-06-13T15:34:22.000-07:00
diff --git a/container/deps/requirements.txt b/container/deps/requirements.txt
@@ -24,13 +24,13 @@ matplotlib
 msgspec
 mypy
 numpy
+nvidia-ml-py==12.575.51
 opentelemetry-api
 opentelemetry-sdk
 pip==25.0.1
 pre-commit
 protobuf==5.27.3
 pydantic==2.7.1
-pynvml
 pyright
 PyYAML
 scikit-learn
