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

2026-04-24 17:20:40 +00:00
parent 0ef1bc45b6
commit f0e5df32ac
1 changed files with 2 additions and 2 deletions
@@ -13,7 +13,7 @@

  # dependencies
  numpy,
-  pynvml,
+  nvidia-ml-py,

  # tests
  einops,
@@ -102,7 +102,7 @@ buildPythonPackage.override { stdenv = effectiveStdenv; } (finalAttrs: {
    torch
  ]
  ++ lib.optionals cudaSupport [
-    pynvml
+    nvidia-ml-py
  ];

  pythonImportsCheck = [