diff --git a/pkgs/development/python-modules/pynvml/default.nix b/pkgs/development/python-modules/pynvml/default.nix new file mode 100644 index 00000000000..c64d723ec32 --- /dev/null +++ b/pkgs/development/python-modules/pynvml/default.nix @@ -0,0 +1,29 @@ +{ lib +, buildPythonPackage +, fetchPypi +, pythonOlder +, cudatoolkit +}: + +buildPythonPackage rec { + pname = "pynvml"; + version = "8.0.4"; + disabled = pythonOlder "3.6"; + + src = fetchPypi { + inherit pname version; + sha256 = "0pfykj1amqh1rixp90rg85v1nj6qmx89fahqr6ii4zlcckffmm68"; + }; + + propagatedBuildInputs = [ cudatoolkit ]; + + doCheck = false; # no tests in PyPi dist + pythonImportsCheck = [ "pynvml" "pynvml.smi" ]; + + meta = with lib; { + description = "Python bindings for the NVIDIA Management Library"; + homepage = https://www.nvidia.com; + license = licenses.bsd3; + maintainers = [ maintainers.bcdarwin ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 5b148263388..438eb0e43ad 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -4943,6 +4943,8 @@ in { pynrrd = callPackage ../development/python-modules/pynrrd { }; + pynvml = callPackage ../development/python-modules/pynvml { }; + pynzb = callPackage ../development/python-modules/pynzb { }; process-tests = callPackage ../development/python-modules/process-tests { };