From f50325059adf7251e8dce7f906a7ebac1e61c6c3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jos=C3=A9=20Luis=20Lafuente?= Date: Tue, 14 Jul 2026 11:29:59 +0200 Subject: [PATCH] python3Packages.nvidia-dlprofviewer: init at 1.8.0 --- .../nvidia-dlprofviewer/default.nix | 58 +++++++++++++++++++ pkgs/top-level/python-packages.nix | 2 + 2 files changed, 60 insertions(+) create mode 100644 pkgs/development/python-modules/nvidia-dlprofviewer/default.nix diff --git a/pkgs/development/python-modules/nvidia-dlprofviewer/default.nix b/pkgs/development/python-modules/nvidia-dlprofviewer/default.nix new file mode 100644 index 000000000000..1ba2c9c8691c --- /dev/null +++ b/pkgs/development/python-modules/nvidia-dlprofviewer/default.nix @@ -0,0 +1,58 @@ +{ + lib, + buildPythonPackage, + fetchurl, + autoAddDriverRunpath, + autoPatchelfHook, + django, + gunicorn, + setuptools, + sqlite, + uvicorn, + whitenoise, +}: + +buildPythonPackage (finalAttrs: { + pname = "nvidia-dlprofviewer"; + version = "1.8.0"; + __structuredAttrs = true; + format = "wheel"; + + src = fetchurl { + url = "https://pypi.nvidia.com/nvidia-dlprofviewer/nvidia_dlprofviewer-${finalAttrs.version}-py3-none-any.whl"; + hash = "sha256-KrfLZ6NdP31qfSfOBl8944ZE2xZ+kgjkyt+QPUblYVA="; + }; + + pythonImportsCheck = [ + "dlprofviewer" + ]; + + nativeBuildInputs = [ + autoAddDriverRunpath + autoPatchelfHook + ]; + + autoPatchelfIgnoreMissingDeps = [ + "libnvidia-ml.so.1" + ]; + + propagatedBuildInputs = [ + django + gunicorn + setuptools + sqlite + uvicorn + whitenoise + ]; + + pythonRelaxDeps = [ + "django" + ]; + + meta = { + description = "NVIDIA DLProf Viewer"; + homepage = "https://docs.nvidia.com/deeplearning/frameworks/dlprof-user-guide/index.html"; + license = lib.licenses.unfree; + maintainers = with lib.maintainers; [ jlesquembre ]; + }; +}) diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 5ebecb4e6a12..a5e3498b76ba 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -11919,6 +11919,8 @@ self: super: with self; { callPackage ../development/python-modules/nvidia-dlprof-pytorch-nvtx { }; + nvidia-dlprofviewer = callPackage ../development/python-modules/nvidia-dlprofviewer { }; + nvidia-ml-py = callPackage ../development/python-modules/nvidia-ml-py { }; nwdiag = callPackage ../development/python-modules/nwdiag { };