python3Packages.onnxscript: use pynvml instead of deprecated nvidia-ml-py (#513986)

2026-04-27 22:07:32 +00:00
parent 6f81182eb8 674601df48
commit 04f0a49315
1 changed files with 3 additions and 2 deletions
@@ -15,7 +15,8 @@
  onnx-ir,
  packaging,
  typing-extensions,
-  pynvml,
+  # cuda-only:
+  nvidia-ml-py,

  # tests
  onnxruntime,
@@ -58,7 +59,7 @@ buildPythonPackage (finalAttrs: {
    typing-extensions
  ]
  ++ lib.optionals cudaSupport [
-    pynvml
+    nvidia-ml-py
  ];

  pythonImportsCheck = [ "onnxscript" ];