diff options
Diffstat (limited to 'pkgs/development/python-modules/pynvml/default.nix')
-rw-r--r-- | pkgs/development/python-modules/pynvml/default.nix | 29 |
1 files changed, 29 insertions, 0 deletions
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 ]; + }; +} |