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

   1 { buildPythonPackage
   2 , blessed
   3 , fetchPypi
   4 , lib
   5 , mockito
   6 , nvidia-ml-py
   7 , psutil
   8 , pytest-runner
   9 , pythonRelaxDepsHook
  10 , pytestCheckHook
  11 }:
  12
  13 buildPythonPackage rec {
  14   pname = "gpustat";
  15   version = "1.0.0";
  16
  17   src = fetchPypi {
  18     inherit pname version;
  19     sha256 = "sha256-WB6P+FjDLJWjIruPA/HZ3D0Xe07LM93L7Sw3PGf04/E=";
  20   };
  21
  22   nativeBuildInputs = [ pythonRelaxDepsHook ];
  23   pythonRelaxDeps = [ "nvidia-ml-py" ];
  24
  25   propagatedBuildInputs = [
  26     blessed
  27     nvidia-ml-py
  28     psutil
  29   ];
  30
  31   checkInputs = [
  32     mockito
  33     pytestCheckHook
  34   ];
  35
  36   pythonImportsCheck = [ "gpustat" ];
  37
  38   meta = with lib; {
  39     description = "A simple command-line utility for querying and monitoring GPU status";
  40     homepage = "https://github.com/wookayin/gpustat";
  41     license = licenses.mit;
  42     maintainers = with maintainers; [ billhuang ];
  43   };
  44 }