python3Packages.xformers: use nvidia-ml-py instead of deprecated pynvml (#513123) (e3c3adb2) · Commits · nix / nixpkgs

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

+2 −2

Original line number	Diff line number	Diff line
		@@ -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 = [