pyrox.dev
1{
2 lib,
3 blessed,
4 buildPythonPackage,
5 fetchPypi,
6 mockito,
7 nvidia-ml-py,
8 psutil,
9 pytestCheckHook,
10 pythonOlder,
11 pythonRelaxDepsHook,
12 setuptools-scm,
13}:
14
15buildPythonPackage rec {
16 pname = "gpustat";
17 version = "1.1.1";
18 format = "setuptools";
19
20 disabled = pythonOlder "3.7";
21
22 src = fetchPypi {
23 inherit pname version;
24 hash = "sha256-wY0+1VGPwWMAxC1pTevHCuuzvlXK6R8dtk1jtfqK+dg=";
25 };
26
27 pythonRelaxDeps = [ "nvidia-ml-py" ];
28
29 nativeBuildInputs = [
30 pythonRelaxDepsHook
31 setuptools-scm
32 ];
33
34 propagatedBuildInputs = [
35 blessed
36 nvidia-ml-py
37 psutil
38 ];
39
40 nativeCheckInputs = [
41 mockito
42 pytestCheckHook
43 ];
44
45 pythonImportsCheck = [ "gpustat" ];
46
47 meta = with lib; {
48 description = "A simple command-line utility for querying and monitoring GPU status";
49 mainProgram = "gpustat";
50 homepage = "https://github.com/wookayin/gpustat";
51 changelog = "https://github.com/wookayin/gpustat/releases/tag/v${version}";
52 license = licenses.mit;
53 maintainers = with maintainers; [ billhuang ];
54 };
55}