nixpkgs/pkgs/tools/system/nvitop/default.nix

41 lines
981 B
Nix

{ lib
, python3Packages
, fetchFromGitHub
}:
python3Packages.buildPythonApplication rec {
pname = "nvitop";
version = "1.3.2";
src = fetchFromGitHub {
owner = "XuehaiPan";
repo = pname;
rev = "refs/tags/v${version}";
hash = "sha256-TunGtNe+lgx/hk8kNtB8yaCdbkiJ3d4JJ8NKB+6urJA=";
};
pythonRelaxDeps = [ "nvidia-ml-py" ];
nativeBuildInputs = with python3Packages; [ pythonRelaxDepsHook ];
propagatedBuildInputs = with python3Packages; [
cachetools
psutil
termcolor
nvidia-ml-py
];
checkPhase = ''
$out/bin/nvitop --help
'';
meta = with lib; {
description = "An interactive NVIDIA-GPU process viewer, the one-stop solution for GPU process management";
homepage = "https://github.com/XuehaiPan/nvitop";
changelog = "https://github.com/XuehaiPan/nvitop/releases/tag/v${version}";
license = licenses.gpl3;
maintainers = with maintainers; [ GaetanLepage ];
platforms = with platforms; linux;
};
}