pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchurl,
5 pythonOlder,
6 numpy,
7 torch,
8}:
9
10buildPythonPackage rec {
11 pname = "nvidia-dlprof-pytorch-nvtx";
12 version = "1.8.0";
13
14 src = fetchurl {
15 url = "https://pypi.nvidia.com/nvidia-dlprof-pytorch-nvtx/nvidia_dlprof_pytorch_nvtx-${version}-py3-none-any.whl";
16 hash = "sha256-IHVCXnkVy3lXw22ISpeMdt5T7UZSzHp8sbWGF/emwGw=";
17 };
18
19 format = "wheel";
20
21 dontBuild = true;
22
23 dependencies = [
24 numpy
25 torch
26 ];
27
28 pythonImportsCheck = [
29 "nvidia_dlprof_pytorch_nvtx"
30 ];
31
32 meta = {
33 description = "NVIDIA DLProf Pytorch NVTX markers";
34 homepage = "https://docs.nvidia.com/deeplearning/frameworks/dlprof-user-guide/index.html";
35 license = lib.licenses.unfree; # NVIDIA Proprietary Software
36 maintainers = with lib.maintainers; [ jherland ];
37 };
38}