nathan/nixpkgs
1
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2026-09-29 19:30:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

python3Packages.nvidia-dlprof-pytorch-nvtx: init at 1.8.0

Browse Source
This commit is contained in:
Johan Herland
2025-07-03 14:15:52 +00:00
parent ce557ce4ac
commit 48db51dbd4
2 changed files with 42 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
pkgs/development/python-modules/nvidia-dlprof-pytorch-nvtx/default.nix Normal file
View File

@@ -0,0 +1,38 @@
{
lib,
buildPythonPackage,
fetchurl,
pythonOlder,
numpy,
torch,
}:
buildPythonPackage rec {
pname = "nvidia-dlprof-pytorch-nvtx";
version = "1.8.0";
src = fetchurl {
url = "https://pypi.nvidia.com/nvidia-dlprof-pytorch-nvtx/nvidia_dlprof_pytorch_nvtx-${version}-py3-none-any.whl";
hash = "sha256-IHVCXnkVy3lXw22ISpeMdt5T7UZSzHp8sbWGF/emwGw=";
};
format = "wheel";
dontBuild = true;
dependencies = [
numpy
torch
];
pythonImportsCheck = [
"nvidia_dlprof_pytorch_nvtx"
];
meta = {
description = "NVIDIA DLProf Pytorch NVTX markers";
homepage = "https://docs.nvidia.com/deeplearning/frameworks/dlprof-user-guide/index.html";
license = lib.licenses.unfree; # NVIDIA Proprietary Software
maintainers = with lib.maintainers; [ jherland ];
};
}

4
pkgs/top-level/python-packages.nix
View File

@@ -10420,6 +10420,10 @@ self: super: with self; {
nvdlib = callPackage ../development/python-modules/nvdlib { };
nvidia-dlprof-pytorch-nvtx =
callPackage ../development/python-modules/nvidia-dlprof-pytorch-nvtx
{ };
nvidia-ml-py = callPackage ../development/python-modules/nvidia-ml-py { };
nwdiag = callPackage ../development/python-modules/nwdiag { };