pyrox.dev
1{
2 lib,
3 blessed,
4 buildPythonPackage,
5 fetchPypi,
6 mockito,
7 nvidia-ml-py,
8 psutil,
9 pytestCheckHook,
10 pythonOlder,
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 = [ "nvidia-ml-py" ];
27
28 nativeBuildInputs = [
29 setuptools-scm
30 ];
31
32 propagatedBuildInputs = [
33 blessed
34 nvidia-ml-py
35 psutil
36 ];
37
38 nativeCheckInputs = [
39 mockito
40 pytestCheckHook
41 ];
42
43 pythonImportsCheck = [ "gpustat" ];
44
45 meta = with lib; {
46 description = "Simple command-line utility for querying and monitoring GPU status";
47 mainProgram = "gpustat";
48 homepage = "https://github.com/wookayin/gpustat";
49 changelog = "https://github.com/wookayin/gpustat/releases/tag/v${version}";
50 license = licenses.mit;
51 maintainers = with maintainers; [ billhuang ];
52 };
53}