pyrox.dev
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
13buildPythonPackage 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}