tjh.dev
1{ lib
2, buildPythonPackage
3, fetchPypi
4, pythonOlder
5, cudatoolkit
6}:
7
8buildPythonPackage rec {
9 pname = "pynvml";
10 version = "8.0.4";
11 disabled = pythonOlder "3.6";
12
13 src = fetchPypi {
14 inherit pname version;
15 sha256 = "0pfykj1amqh1rixp90rg85v1nj6qmx89fahqr6ii4zlcckffmm68";
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}