pyrox.dev
1{ lib
2, buildPythonPackage
3, fetchPypi
4, pythonOlder
5, cudatoolkit
6}:
7
8buildPythonPackage rec {
9 pname = "pynvml";
10 version = "11.4.1";
11 disabled = pythonOlder "3.6";
12
13 src = fetchPypi {
14 inherit pname version;
15 sha256 = "b2e4a33b80569d093b513f5804db0c7f40cfc86f15a013ae7a8e99c5e175d5dd";
16 };
17
18 propagatedBuildInputs = [ cudatoolkit ];
19
20 doCheck = false; # no tests in PyPi dist
21 pythonImportsCheck = [ "pynvml" "pynvml.smi" ];
22
23 meta = with lib; {
24 description = "Python bindings for the NVIDIA Management Library";
25 homepage = "https://www.nvidia.com";
26 license = licenses.bsd3;
27 maintainers = [ maintainers.bcdarwin ];
28 };
29}