python3Packages.xformers: use nvidia-ml-py instead of deprecated pynvml (#513123)

2026-04-24 22:29:41 +00:00
parent 5dd153372d f0e5df32ac
commit e3c3adb2a2
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 = [