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