Commit 674601df authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python3Packages.onnxscript: use pynvml instead of deprecated nvidia-ml-py

parent f6770b8b

pkgs/development/python-modules/onnxscript/default.nix

+3 −2
Original line number Diff line number Diff line
@@ -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" ];