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